size() 和 shape函数的用法

作用

• size() 和 shape 的作用类似, 都是查看数据的类型

• 注意使用 size() 函数需要加括号, shape 不需要加括号, 加括号会报错

返回结果

• 函数返回一个列表, 第一个表示 batch_size, 后两个表示每一个batch 的行和列

示例

import torch

a = torch.tensor([[[1, 1], [2, 2], [3, 3]], 
                 [[4, 4], [5, 5], [6, 6]]])
print(a.size()[-2:])
print(a.size())
print(a.shape)

>>torch.Size([3, 2])
>>torch.Size([2, 3, 2])
>>torch.Size([2, 3, 2])