rocAL_pybind.amd.rocal.plugin.pytorch.ROCALAudioIterator Class Reference#
rocAL_pybind.amd.rocal.plugin.pytorch.ROCALAudioIterator Class Reference
rocAL iterator for audio tasks for PyTorch The Tensors that are returned by the iterator will be owned by rocAL and would be valid until next iteration. More...
Inheritance diagram for rocAL_pybind.amd.rocal.plugin.pytorch.ROCALAudioIterator:
Data Fields | |
loader | |
device | |
device_id | |
output | |
iterator_length | |
max_shape | |
batch_size | |
output_list | |
labels_size | |
output_memory_type | |
output_tensor_list | |
num_roi_dims | |
roi_array | |
labels_tensor | |
Detailed Description
rocAL iterator for audio tasks for PyTorch The Tensors that are returned by the iterator will be owned by rocAL and would be valid until next iteration.
- Parameters
-
pipeline The rocAL pipeline to use for processing data. tensor_dtype Data type of the output tensors. size Number of samples in the epoch (Usually the size of the dataset). auto_reset Whether the iterator resets itself for the next epoch or it requires reset() to be called separately. device The device to use for processing - CPU / GPU device_id The ID of the device to use
The documentation for this class was generated from the following file:
- /home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-rocal/checkouts/docs-6.1.2/rocAL_pybind/amd/rocal/plugin/pytorch.py