def _shape(input, out_type=np.int32, name=None):
    return ops.convert_to_tensor(ops.convert_to_tensor(input).shape).astype(out_type)
def _shape(input, out_type=np.int32, name=None):
    return ops.convert_to_tensor(ops.convert_to_tensor(input).shape).astype(out_type)