PyTorch中创建Tensor的几种常用函数

最新推荐文章于 2025-04-16 19:27:41 发布

原创最新推荐文章于 2025-04-16 19:27:41 发布 · 444 阅读

1 ·

本内容遵循CC 4.0 BY-SA版权协议

收录于

博客聚焦于PyTorch中创建Tensor的常用函数，在信息技术领域，Tensor是深度学习等应用的重要数据结构，了解创建它的常用函数对相关开发和研究有重要意义。

import torch

mean = torch.tensor([10,20,30],dtype=torch.float32)
a = torch.normal(mean, 1.0) #均值分别为[10,20,30], std=1, size与均值保持一致
b = torch.randn(2,3) #标准正态分布 mean=0, std=1, size=(2,3)
c = torch.rand(1,3) #区间[0,1)随机采样,size=(1,3)
d = torch.randint(5,10,(1,5)) #区间[5,10)随机采样整数, size=(1,5)
e = torch.randperm(5) #生成0到n-1的随机排序