(x, axis=1, keepdims=True)
return np.argmin(x, axis=axis).reshape(new_shape)
Calls numpy argmin with keepdims.
Calls numpy argmin with keepdims.
Calls numpy argmin with keepdims .
def argmin(x, axis=1, keepdims=True):
        """
        Calls numpy argmin with keepdims.
        """
        new_shape = list(x.shape)
        new_shape[axis] = 1
        new_shape = tuple(new_shape)
        return np.argmin(x, axis=axis).reshape(new_shape)
def argmin ( x , axis = 1 , keepdims = True ) : new_shape = list ( x . shape ) new_shape [ axis ] = 1 new_shape = tuple ( new_shape ) return np . argmin ( x , axis = axis ) . reshape ( new_shape )