Function ref

dali/test/python/experimental_mode/test_tensor.py:400–401

(batch)

)
def test_batch_to_tensor(src_device, target_device, force_copy, dtype, layout, data_factory):
    def ref(batch):
        return ndd.stack(*ndd.pad(batch).tensors, axis_name="N").evaluate()

    pad = data_factory == ragged  # only apply padding when necessary