PyTorch常用数学运算速查表（29种常用张量运算函数）

PyTorch常用数学运算速查表

PyTorch官方文档共给出90种PyTorch张量运算函数，这里以表格的形式列出常用的29种运算，便于速查。

函数	作用
add()	Tensor中每个元素同加一个标量，或与另一个Tensor逐元素相加
mul()	Tensor中每个元素同乘一个标量，或与另一个Tensor逐元素相乘
div()	Tensor中每个元素同除一个标量，或与另一个Tensor逐元素相除
dot()	对两个Tensor向量进行“点积”
mm()	Tensor矩阵乘Tensor矩阵
mv()	Tensor矩阵乘Tensor向量
fmod()	Tensor中每个元素与一个标量的除法余数，相当于Python中的%操作
remainder()	Tensor中每个元素与一个标量的除法余数，相当于Python中的%操作
abs()	对Tensor中的每个元素取绝对值，并返回
ceil()	对Tensor中的每个元素向上取整
floor()	对Tensor中的每个元素向下取整
clamp()	对Tensor中的每个元素取上下限
round()	对Tensor中的每个元素取最近的整数
frac()	返回Tensor中每个元素的分数部分
neg()	对Tensor中的每个元素取负
reciprocal()	对Tensor中的每个元素取倒数
log()	返回一个张量，包含Tensor中每个元素的自然对数
pow()	对Tensor中的每个元素同取一个标量幂值，或采用另外一个Tensor的对应元素取幂
exp()	返回一个张量，包含Tensor中每个元素的指数
sigmoid()	返回一个张量，包含Tensor中的每个元素的sigmoid值
sign()	返回一个张量，包含Tensor中每个元素的正负值
sqrt()	返回一个张量，包含Tensor中每个元素的平方根
dist()	返回两个Tensor的范数
mean()	返回Tensor中所有元素的均值
norm()	返回Tensor的范数值
prod()	返回Tensor的所有元素之积
sum()	返回Tensor的 所有元素之和
max()	返回Tensor的所有元素的最大值
min()	返回Tensor的所有元素的最小值