tensor

作者: dc_3 | 来源:发表于2022-02-16 11:25 被阅读0次

一维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

相关文章

网友评论

      本文标题:tensor

      本文链接:https://www.haomeiwen.com/subject/grjzkrtx.html