Method reshape

ot/backend.py:2216–2217

(self, a, shape)

        return torch.stack(arrays, dim=axis)

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

    def seed(self, seed=None):
        if seed is None: