上QQ阅读APP看书,第一时间看更新
去QQ阅读APP
上一章
目录
下一章
4.4 PyTorch的Reduction操作
Reduction运算的特点是它往往对一个Tensor内的元素做归约操作,如torch.max找极大值,torch.cumsum计算累加。另外它还提供了dim参数来指定沿矩阵的哪个维度执行操作。
上一章
目录
下一章