tensor

一维tensor:向量

  • shape: torch.Size([10]) (10个元素举例)
  • data
    tensor([-0.1345, -0.6423, 1.0208, -0.9232, 1.1923, -1.1397, 1.4297, -0.5168,-2.6687, -0.6039])(注意这里是一个中括号

二维tensor:矩阵

  • shape:torch.Size([1,10])
  • data:
    tensor([[-1.2320, -0.4606, -0.4050, 1.2633, 0.3945, -0.0284, 0.8834, 0.7251, 1.1978, -1.1629]])(**注意这里是两个中括号)

三维tensor:RGB图像

  • shape:torch.Size([3,10,10])

四维tensor:

Conv2d等卷积函数的输入tensor:分别是 batchsize,channel,高度,宽度。
*小技巧:通过数中括号的个数,就能得出数据的维度。

tensor对象属性

图片.png
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容