创建矩阵
在 PyTorch 中，我们可以使用以下方法来创建矩阵：
-
使用列表或 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) -
使用特定的函数创建：
# 创建全零矩阵 tensor3 = torch.zeros(2, 3) # 创建全一矩阵 tensor4 = torch.ones(2, 3) # 创建随机矩阵（均匀分布） tensor5 = torch.rand(2, 3) # 创建随机矩阵（正态分布） tensor6 = torch.randn(2, 3) -
使用其他矩阵的形状创建：文章来源：https://www.toymoban.com/news/detail-518376.html
# 创建与指定矩阵形状相同的全零矩阵 tensor7 = torch.zeros_like(tensor1) # 创建与指定矩阵形状相同的全一矩阵 tensor8 = torch.ones_like(tensor1) # 创建与指定矩阵形状相同的随机矩阵（均匀分布） tensor9 = torch.rand_like(tensor1) # 创建与指定矩阵形状相同的随机矩阵（正态分布） tensor10 = torch.randn_like(tensor1)
以上方法可以根据具体的需求来创建不同形状、不同数值的矩阵。需要注意的是，这些函数创建的张量的默认数据类型是 torch.float32。如果需要使用其他数据类型，可以通过 dtype 参数指定。
看到这里点个赞吧(*^▽^*)文章来源地址https://www.toymoban.com/news/detail-518376.html
到了这里，关于pytorch中创建矩阵的诸多方法的文章就介绍完了。如果您还想了解更多内容，请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章，希望大家以后多多支持TOY模板网！
