(self, a, axis=None)
| 2029 | return torch.sum(a, axis, keepdim=keepdims) |
| 2030 | |
| 2031 | def cumsum(self, a, axis=None): |
| 2032 | if axis is None: |
| 2033 | return torch.cumsum(a.flatten(), 0) |
| 2034 | else: |
| 2035 | return torch.cumsum(a, axis) |
| 2036 | |
| 2037 | def max(self, a, axis=None, keepdims=False): |
| 2038 | if axis is None: |