pytorch学习手册【二】

  九、Reduction Ops（规约/简化操作） torch.argmax(input, dim=None, keepdim=False) torch.argmin(input, dim=None, keepdim=False)   torch.cumprod(input, dim, dtype=None) → Tensor torch.cumsum(input, dim, out=None,

>>阅读原文<<