深度学习框架_PyTorch_torch.size()函数
作用：查看Tensor的维度。具体例子如下：>>> import torch>>> a = torch.Tensor([[[1,1],[2,2]],[[3,3],[4,4]]])>>> a.size()torch.Size([2, 2, 2])
无心曲奇
他的专栏
他的归档
热门文章
