f08b65795c016562eb9dac1daffde50cec4d448c,tensorly/backend/mxnet_backend.py,,argmin,#Any#Any#,118

Before Change

    return nd.argmax(data, axis).astype("int32").asscalar()

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

def abs(tensor, **kwargs):
    if isinstance(tensor, nd.NDArray):
        return nd.abs(tensor, **kwargs)

After Change

        return res

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

Instances