pytorch中创建矩阵的诸多方法

创建矩阵

在 PyTorch 中，我们可以使用以下方法来创建矩阵：

1. 使用列表或 NumPy 数组创建：

   import torch
   import numpy as np
   
   # 使用列表创建矩阵
   lst = [[1, 2, 3], [4, 5, 6]]
   tensor1 = torch.tensor(lst)
   
   # 使用 NumPy 数组创建矩阵
   arr = np.array([[1, 2, 3], [4, 5, 6]])
   tensor2 = torch.from_numpy(arr)
   

2. 使用特定的函数创建：

   # 创建全零矩阵
   tensor3 = torch.zeros(2, 3)
   
   # 创建全一矩阵
   tensor4 = torch.ones(2, 3)
   
   # 创建随机矩阵（均匀分布）
   tensor5 = torch.rand(2, 3)
   
   # 创建随机矩阵（正态分布）
   tensor6 = torch.randn(2, 3)
   

3. 使用其他矩阵的形状创建：

   # 创建与指定矩阵形状相同的全零矩阵
   tensor7 = torch.zeros_like(tensor1)
   
   # 创建与指定矩阵形状相同的全一矩阵
   tensor8 = torch.ones_like(tensor1)
   
   # 创建与指定矩阵形状相同的随机矩阵（均匀分布）
   tensor9 = torch.rand_like(tensor1)
   
   # 创建与指定矩阵形状相同的随机矩阵（正态分布）
   tensor10 = torch.randn_like(tensor1)
   

以上方法可以根据具体的需求来创建不同形状、不同数值的矩阵。需要注意的是，这些函数创建的张量的默认数据类型是 torch.float32。如果需要使用其他数据类型，可以通过 dtype 参数指定。
看到这里点个赞吧(*^▽^*)