PyTorch 的 Tensor 是张量(多维数组)的数学抽象,与 Numpy 数组类似。下面是如何使用 PyTorch Tensor 的一些示例:
- 创建 Tensor:
import torch
# 创建一个数字为 1 的 1 维 Tensor
a = torch.tensor(1)
# 创建一个数组为 [1, 2, 3] 的 1 维 Tensor
b = torch.tensor([1, 2, 3])
# 创建一个形状为 (3, 3),数字全为 0 的 2 维 Tensor
c = torch.zeros(3, 3)
# 创建一个形状为 (3, 3),数字全为 1 的 2 维 Tensor
d = torch.ones(3, 3)
# 创建一个形状为 (3, 3),数字随机的 2 维 Tensor
e = torch.rand(3, 3)