- Python 深度学习
- 吕云翔 刘卓然 关捷雄等编著
- 69字
- 2025-02-28 02:11:36
4.4 PyTorch的Reduction操作
Reduction运算的特点是它往往对一个Tensor内的元素做归约操作,如torch.max找极大值,torch.cumsum计算累加。另外它还提供了dim参数来指定沿矩阵的哪个维度执行操作。


Reduction运算的特点是它往往对一个Tensor内的元素做归约操作,如torch.max找极大值,torch.cumsum计算累加。另外它还提供了dim参数来指定沿矩阵的哪个维度执行操作。