batch.size()[0]与batch.size()[1]的意思

跑程序遇到这个不太理解,于是自己敲了一遍代码:

 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)

猜你喜欢

转载自blog.csdn.net/Viviane_2022/article/details/128502602
今日推荐