在PyTorch中创建张量有多种方法，最常用的方法包括

lewis 2年前 (2023-10-28) 阅读数 11 #技术

在PyTorch中创建张量有多种方法，最常用的方法包括：

importtorch

tensor=torch.tensor([1,2,3,4,5])

使用torch.ones()函数或torch.zeros()函数：创建全一或全零的张量。

importtorch

ones_tensor=torch.ones(2,3)
zeros_tensor=torch.zeros(3,2)

使用torch.randn()函数：创建一个符合正态分布的随机张量。

importtorch

random_tensor=torch.randn(2,3)

使用torch.arange()函数或torch.linspace()函数：创建一个等差数列或等分数列的张量。

importtorch

arange_tensor=torch.arange(1,10,2)
linspace_tensor=torch.linspace(1,10,5)

除了以上方法外，还可以根据自己的需求使用其他方法来创建张量，如torch.eye()函数用于创建单位矩阵张量等。

本文仅代表作者观点，不代表博信信息网立场。