1.torch. from_numpy
从np.ndarray创建一个张量
2.torch. linspace(start, end, step)
返回开始和结束之间等间隔点的一维张量,step为点的个数
3.torch.rand(*sizes)
返回一个张量,包含了从区间[0,1)的均匀分布中抽取的一组随机数,形状由可变参数sizes定义
4.torch.randn(*sizes)
返回一个张量,包含从标准正态分布中(均值为0,方差为1)中抽取的一组随机数,形状由可变参数sizes定义
5.torch.arange(start, end, step)
返回一个一维向量,长度为(end-start)/step
6.torch.ones() / torch.zeros()
返回由标量值1或0填充的张量,其形状由可变参数大小定义。
7.torch.normal(means, std)
返回一个张量,包含从给定参数means,std的离散正态分布中抽取随机数。 均值means是一个张量,包含每个输出元素相关的正态分布的均值。 std是一个张量,包含每个输出元素相关的正态分布的标准差。 均值和标准差的形状不须匹配,但每个张量的元素个数须相同。
8.torch.cat(inputs, dimension=0)
Pytorch基本操作
最新推荐文章于 2024-08-05 15:53:03 发布