(三)对张量的基础操作

2020-02-02  本文已影响0人  zelda2333

数据类型

python PyTorch
Int IntTensor of size()
float FloatTensor of size()
Int array IntTensor of size[d1, d2,...]
float array FloatTensor of size[d1, d2,...]
string ---

判断数据类型

dimension 0 / rank0(标量)


1.是0维,但[1.]是1维,长度为1 的Tensor
loss通常是标量


Dim1 / rank1(向量)

Bias, Linear Input 常为Dim 1

Dim 2


带有batch的Linear Input 常为2维,第一张照片第二张照片[1, 784]

Dim 3


带有batch的RNN Input 常为3维,[word, sentence, 100]

Dim 4


CNN[b, c, h, w]
b:几张照片



shape返回tensor的形状;
numel返回占用内存大小(4704 = 2 * 3 * 28 * 28)

上一篇下一篇

猜你喜欢

热点阅读