def _size(input, out_type=np.int32, name=None):
    return np.asarray(np.prod(ops.convert_to_tensor(input).shape), dtype=out_type)
def _size(input, out_type=np.int32, name=None):
    return np.asarray(np.prod(ops.convert_to_tensor(input).shape), dtype=out_type)