PyTorch获取Tensor大小，size与shape的区别

Posted 2022-08-23 算法与编程之美

问题

Tensor是PyTorch的重要概念之一，获取Tensor的大小的方法主要有两种，其一是size，其二是shape，那么两种方法的区别是什么呢？

方法

import torch

a = torch.rand(size=(3, 224, 224))
# shape是Tensor的一个属性
print(a.shape) # torch.Size([3, 224, 224])

# size()是Tensor的一个方法
print(a.size()) # torch.Size([3, 224, 224])

# 给定参数获取不同的维度size
print(a.size(0)) # 3
print(a.size(1)) # 224 
print(a.size(2)) # 224

结语