a = torch.tensor([[1,2,3],[4,5,6]])
a.size()
torch.Size([2, 3])
a.shape
torch.Size([2, 3])
看起来,shape是一个Tensor的属性,而size()是Tensor的函数。
一般我们都是用a.size()来获得Tensor的形状
a = torch.tensor([[1,2,3],[4,5,6]])
a.size()
torch.Size([2, 3])
a.shape
torch.Size([2, 3])
看起来,shape是一个Tensor的属性,而size()是Tensor的函数。
一般我们都是用a.size()来获得Tensor的形状