tensor

2022-02-16  本文已影响0人  dc_3

一维tensor:向量

二维tensor:矩阵

三维tensor:RGB图像

四维tensor:

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

tensor对象属性

图片.png
上一篇 下一篇

猜你喜欢

热点阅读