(self, a, axis=None)
| 2115 | return torch.searchsorted(a, v, right=right) |
| 2116 | |
| 2117 | def flip(self, a, axis=None): |
| 2118 | if axis is None: |
| 2119 | return torch.flip(a, tuple(i for i in range(len(a.shape)))) |
| 2120 | if isinstance(axis, int): |
| 2121 | return torch.flip(a, (axis,)) |
| 2122 | else: |
| 2123 | return torch.flip(a, dims=axis) |
| 2124 | |
| 2125 | def outer(self, a, b): |
| 2126 | return torch.outer(a, b) |