PyTorch是一个广泛使用的深度学习框架,它提供了许多强大的工具和功能来处理和操作张量(Tensor)。在本教程中,我们将深入探讨PyTorch张量的使用方法和一些常见操作。
- 张量的创建
在PyTorch中,可以使用torch.Tensor()函数来创建一个张量。张量可以是标量(0维张量)、向量(1维张量)、矩阵(2维张量)或更高维度的数组。下面是一些创建张量的示例代码:
import torch
# 创建一个标量(0维张量)
scalar_tensor = torch.tensor(5)
print(scalar_tensor