'calls numpy argmin with keepdims'

@staticmethod
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)
