Function reshape_pipe

dali/test/python/operator_2/test_reshape.py:329–331

(shapes, src_dims=None, rel_shape=None)

@pipeline_def
def reshape_pipe(shapes, src_dims=None, rel_shape=None):
    data = fn.external_source(lambda: get_data(shapes), batch=True, device="cpu")
    return fn.reshape(data, src_dims=src_dims, rel_shape=rel_shape)


def _testimpl_reshape_src_dims_arg(src_dims, rel_shape, shapes, expected_out_shapes):