(self, a, return_inverse=False)
| 2202 | return torch.diag(a, diagonal=k) |
| 2203 | |
| 2204 | def unique(self, a, return_inverse=False): |
| 2205 | return torch.unique(a, return_inverse=return_inverse) |
| 2206 | |
| 2207 | def logsumexp(self, a, axis=None, keepdims=False): |
| 2208 | if axis is not None: |