Pytorch入门教程（二）：创建Tensor

最新推荐文章于 2024-07-09 08:56:16 发布

王氏小明

最新推荐文章于 2024-07-09 08:56:16 发布

阅读量4.6k

点赞数 1

分类专栏： Pytorch

本文链接：https://blog.csdn.net/weixin_43472830/article/details/94163060

版权

本文详细介绍了PyTorch中创建Tensor的各种方法，包括从numpy导入数据、直接创建二维和多维Tensor、利用均匀分布和正态分布生成随机数、设置全1和全0阵以及对角阵，并探讨了索引操作和大小变换，是PyTorch初学者的实用指南。

摘要由CSDN通过智能技术生成

1. 从numpy中导入数据

a = np.array([2, 5.5])
print(a)
b = torch.from_numpy(a)
print(b)

结果：

[2. 5.5]
tensor([2.0000, 5.5000], dtype=torch.float64)

2. 利用tensor直接创建

b = torch.tensor([[2, 3, 4], [4, 5, 6]])
print(b)

tensor([[2, 3, 4],
[4, 5, 6]])

3 均匀分布

a = torch.rand(3, 3)  # [0,1]之间均匀分布
print(a)
b = torch.rand_like(a)  # 创建一个与a shape相同的数组
print(b)
c = torch.randint(1, 10, (3, 3))  # [0, 10)之间，3*3形状的整数
print(c)

tensor([[0.0789, 0.7449, 0.9799],
[0.7051, 0.3074, 0.7543],
[0.8140, 0.4832, 0.3603]])
tensor([[0.7311, 0.5235, 0.7582],
[0.6360, 0.4095, 0.7638],
[0.5338, 0.1844, 0.5473]])
tensor([[5, 9, 9],
[4, 1, 1],
[9, 5, 5]])