MCPcopy Create free account
hub / github.com/NVIDIA/DALI / asnumpy

Function asnumpy

dali/test/python/experimental_mode/test_batch.py:25–29  ·  view source on GitHub ↗
(batch_or_tensor)

Source from the content-addressed store, hash-verified

23
24
25def asnumpy(batch_or_tensor):
26 if isinstance(batch_or_tensor, ndd.Batch):
27 return [test_tensor.asnumpy(t) for t in batch_or_tensor.tensors]
28 else:
29 return test_tensor.asnumpy(batch_or_tensor)
30
31
32@eval_modes()

Callers 15

test_batch_constructionFunction · 0.70
test_batch_construction_from_list_of_dali_tensorsFunction · 0.70
test_as_batch_does_not_eagerly_create_storageFunction · 0.70
test_batch_construction_from_list_of_torch_gpu_tensors_with_layoutFunction · 0.70
test_batch_construction_from_list_of_torch_gpu_tensors_dtype_fallbackFunction · 0.70
test_batch_construction_from_list_of_torch_cpu_tensorsFunction · 0.70
test_batch_construction_with_conversionFunction · 0.70
test_batch_subscript_broadcastFunction · 0.70
test_batch_partial_sliceFunction · 0.70
test_batch_sliceFunction · 0.70
test_batch_subscript_per_sampleFunction · 0.70
test_batch_to_gpuFunction · 0.70

Calls

no outgoing calls

Tested by

no test coverage detected