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 参数指定。
看到这里点个赞吧(*^▽^*)