Python 深度学习
上QQ阅读APP看书,第一时间看更新

4.4 PyTorch的Reduction操作

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