本文详细介绍了 PyTorch 中张量(Tensor)的 shape 属性及其相关属性,包括 tensor.size()、tensor.ndim 和 tensor.numel()。通过示例代码,展示了如何获取和使用这些属性来了解张量的维度信息和元素总数。这些属性在处理张量时非常有用,特别是在需要进行形状操作或了解张量结构时。文章还解释了 shape 属性返回的 torch.Size 对象,并提供了详细的示例和代码片段,帮助读者更好地理解和应用这些概念。 #Pytorch #张量 #Tensor #张量维度