9d6281d3b2f472e5e130dbf88078c433d0795f46,geomstats/_backend/pytorch/__init__.py,,all,#Any#Any#,228

Before Change

def all(x, axis=None):
    if axis is None and torch.is_tensor(x):
        return x.bool().all()
    numpy_result = _np.array(_np.all(_np.array(x), axis=axis))
    return torch.from_numpy(numpy_result)


def get_slice(x, indices):

After Change

    if isinstance(axis, int):
        return torch.all(x.bool(), axis)
    if len(axis) == 2:
        axis = list(axis)
        for i_axis, one_axis in enumerate(axis):
            if one_axis < 0:
                axis[i_axis] = ndim(x) + one_axis
        return torch.all(
            torch.all(x.bool(), axis[1]), axis[0])
    else:
        raise NotImplementedError(

Instances