1. 简介

tensor(张量)是 PyTorch 中的多维数组,类似与 Numpy 中的 ndarray 。

2. 基本功能

2.1 函数原型

  • torch.empty(*sizes, out=None) → tensor
  • torch.rand(*sizes, out=None) → tensor
  • torch.randn(*sizes, out=None) → tensor
  • torch.randperm(n, out=None) → longtensor

2.2 举例

1
2
3
4
x = torch.empty(5,3)    # 创建 5*3 的未初始化张量
x = torch.rand(5,3) # 创建 5*3 的随机初始化张量([0,1)的均匀分布)
x = torch.randn(5,3) # 创建 5*3 的随机初始化张量(均值为 0 方差为 1 的标准正态分布)
x = torch.randperm(5) # 创建从 0 到 5-1 的随机整数排列