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,
相关文章
相关标签/搜索