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的所有元素的最小值 |



