Torch下的常用函数:

Function

作用

torch.zeros(sizes, out=None)

生成尺寸为 size 的 0 填充向量

torch.ones(sizes, out=None)

生成尺寸为 size 的 1 填充向量

torch.eyes(n, m=None, out=None)

生成对角线全1,其余部分全 0 的二维向量

torch.bmm(input, mat2, out=None)

input,mat2必须是 3维 的,且其第 1 维都一样

torch.​​matmul​​(input, other, out=None)

处理维度不同的 tensor 间的相乘操作(见扩展链接)

torch.LongTensor()

默认生成 64位 整数,dtype 为 torch.int64

torch.FloatTensor()

默认生成 32位 浮点数,dtype 为 torch.float32

torch.rand(sizes, out=None)

返回一个张量,包含了从区间 [0, 1) 的均匀分布中抽取的一组随机数

troch.randn(sizes, out=None)

返回一个张量,包含了从标准正态分布(均值为0,方差为1,即高斯白噪声)中抽取的一组随机数

torch.randint(low=0, high, size, out=None, dtype=None)

返回一个张量,该张量填充在 [low, high) 之间均匀生成的随机整数

torch.normal(means, std, out=None)

返回满足正态分布的张量,means 和 std 分别给出均值和标准差(见扩展链接)