463b5539fb1ca42fafd4791c539a53d1fc49e9ae,tensorly/backend/mxnet_backend.py,,argmax,#Any#Any#,115

Before Change

        return numpy.max(tensor, *args, **kwargs)

def argmax(data=None, axis=None):
    return nd.argmax(data, axis).astype("int32").asscalar()

def argmin(data=None, axis=None):
    return nd.argmin(data, axis).astype("int32").asscalar()

After Change

        return numpy.max(tensor, *args, **kwargs)

def argmax(data=None, axis=None):
    res = nd.argmax(data, axis)
    if res.shape == (1,):
        return res.astype("int32").asscalar()
    else:
        return res

def argmin(data=None, axis=None):
    res = nd.argmin(data, axis)
    if res.shape == (1,):
        return res.astype("int32").asscalar()

Instances