torch.size ():查看当前Tensor的维度,用法也很简单:
终端进入Python环境
>>>import torch
>>>a = torch.Tensor([[1, 2, 3], [4, 5, 6]])
>>>a.size()
(2, 3)
torch.size ():查看当前Tensor的维度,用法也很简单:
终端进入Python环境
>>>import torch
>>>a = torch.Tensor([[1, 2, 3], [4, 5, 6]])
>>>a.size()
(2, 3)