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

Class PipelineTensorAdapter

dali/python/nvidia/dali/_tensor_formatting.py:149–153  ·  view source on GitHub ↗

Adapter for TensorCPU/TensorGPU. Implements TensorAdapter protocol.

Source from the content-addressed store, hash-verified

147
148
149class PipelineTensorAdapter(_BasePipelineAdapter):
150 """Adapter for TensorCPU/TensorGPU. Implements TensorAdapter protocol."""
151
152 def to_numpy(self, obj):
153 return np.array(self.to_cpu(obj))
154
155
156class PipelineBatchAdapter(_BasePipelineAdapter):

Callers 2

_tensor_to_stringFunction · 0.90
format_tensorFunction · 0.85

Calls

no outgoing calls

Tested by

no test coverage detected