如何判断张量(tensor)的维度呢?

利用python随机生成一个张量

a=torch.randn(2,3,1)

输出:

基于上述数据可以得到以下观察,首先确定这是一个三维向量,这个可以根据最左边方括号([ )的数量来确定。

那么它的形状必然是(a,b,c)的形式。

看左边的第一个方括号([ )里面包含了几个元素,这里的元素是以([ ])为一组,因此a=2;再看第二个方括号([ )里包含了几个元素,这里的元素是以([ ])为一组,因此b=3;最后看第三个方括号([  )里面的元素个数,由于这是最后一维数据了,因此直接数这组括号里面的元素个数,因此c=1.因此a.size()=([2,3,1])

为了更好的理解这个概念,我们利用python再随机生成一组数据

b=torch.randn(2,1,3)

输出:

仍然按照上述方法进行判断,最左边的方括号里面包含2组次左方括号,a=2;次左方括号里面包含一组次次左方括号,b=1;最后一组方括号里面包含3个元素,c=3.因此,b.size()=([2,1,3])

评论 6
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值