张量的shape

torch.randint返回一个张量,从标准正态分布随机抽取一组数。
如下图是用torch.randint产生张量,利用shape属性访问各个维度的维数

1、一维张量

在这里插入图片描述

2、二维张量

在这里插入图片描述
如图是一个3*4的二维数组,x.shape[0]表示行数,x.shape[1]表示列数

3、三维张量

在这里插入图片描述
x是一个234 的三维数组,包含2个3*4的二维数组 ,x.shape[0]代表包含二维数组的个数,x.shape[1]表示二维数组的行数,x.shape[2]表示二维数组的列数。

4、总结

[a,b…c]中有N个数,也就是N维数组。表示a个N-1维数组,每个N-1维数组中农有b个N-2个数组…一维数组上有c个数,层层递减,分别对应shape[0],shape[1]…shape[N],也即是每个维度上的大小。
图上看的话,几个中括号就是几维,大维包小维,也就是对应层层拆。

张量就是多维数组,矩阵只是二维数组,没有多维矩阵,因为矩阵只有两个轴,别记混了。
附录
在这里插入图片描述
如上,是个四维矩阵,从最左边数,有4个[。从最低维开始看,也就是只有一个括号[]的,比如
在这里插入图片描述

这里面有两个数,那么对应最后一个2,也就是shape[3],也就是一维里面有2个
然后再往上数一维,有两个[[ ]],对应倒数第二个2,也就是shape[2],也就是二维里面有两行,这里是计数了(也就是一维了)。
在这里插入图片描述
在这里插入图片描述
同理,三维表示有三个[[[ ]]],然后可以看出是三个二维,也就是对应的shape[1]的3。
四维找[[[[ ]]]],可以看出整个中有2个三维,对应shape[0]的2。
参考:https://blog.csdn.net/liuweiyuxiang/article/details/79384435
https://blog.csdn.net/Hningning/article/details/102962402
https://blog.csdn.net/weixin_43050017/article/details/108089406

  • 2
    点赞
  • 16
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值