Method stack

ot/backend.py:2213–2214

(self, arrays, axis=0)

            return torch.logsumexp(a, dim=tuple(range(len(a.shape))), keepdim=keepdims)

    def stack(self, arrays, axis=0):
        return torch.stack(arrays, dim=axis)

    def reshape(self, a, shape):
        return torch.reshape(a, shape)