Method argmin

ot/backend.py:2152–2153

(self, a, axis=None)

        return torch.argmax(a, dim=axis)

    def argmin(self, a, axis=None):
        return torch.argmin(a, dim=axis)

    def mean(self, a, axis=None):
        if axis is not None: