12 Torch下的常用函数

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.IntTensor() 默认生成 32位 整数,dtype 为 torch.int32
torch.FloatTensor() 默认生成 32位 浮点数,dtype 为 torch.float32
torch.DoubleTensor() 默认生成 64位 浮点数,dtype 为 torch.float64
torch.LongTensor() 默认生成 64位 整数,dtype 为 torch.int64
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 分别给出均值和标准差(见扩展链接)
tensor.numel() 返回 tensor 中元素的个数

矩阵乘法:https://blog.csdn.net/xinjieyuan/article/details/114268359?spm=1001.2101.3001.6650.1&utm_medium=distribute.pc_relevant.none-task-blog-2~default~CTRLIST~default-1.no_search_link&depth_1-utm_source=distribute.pc_relevant.none-task-blog-2~default~CTRLIST~default-1.no_search_link

posted @ 2021-11-04 15:44  SethDeng  阅读(114)  评论(0编辑  收藏  举报