跑程序遇到这个不太理解,于是自己敲了一遍代码:
batch.size()[0]表示第0维的维度,
如x = torch.Tensor(2,6),第0维的维度就是2
如x = torch.Tensor(2,6),第1维的维度就是6
import torch
x = torch.Tensor(2,6)
print(x)
x =x.size()[1]
print(x)
结果如下:
import torch
x = torch.Tensor(2,6)
print(x)
x =x.size()[0]
print(x)