rocAL_pybind.amd.rocal.plugin.pytorch.ROCALGenericIterator Class Reference#
rocAL_pybind.amd.rocal.plugin.pytorch.ROCALGenericIterator Class Reference
Iterator for processing data. More...
Inheritance diagram for rocAL_pybind.amd.rocal.plugin.pytorch.ROCALGenericIterator:
Detailed Description
Iterator for processing data.
@param pipeline The rocAL pipeline to use for processing data. @param tensor_layout The layout of the output tensors @param reverse_channels Whether to reverse the order of color channels. @param multiplier Multiplier values for color normalization. @param offset Offset values for color normalization. @param tensor_dtype Data type of the output tensors. @param display Whether to display images during processing @param device The device to use for processing @param 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