rocAL_pybind.amd.rocal.plugin.pytorch.ROCALAudioIterator Class Reference

• rocAL_pybind
• amd
• rocal
• plugin
• pytorch
• ROCALAudioIterator

Public Member Functions | Data Fields

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:

Public Member Functions
def	__init__ (self, pipeline, tensor_dtype=types.FLOAT, size=-1, auto_reset=False, device="cpu", device_id=0)
def	next (self)
def	__next__ (self)
def	reset (self)
def	__iter__ (self)
def	__len__ (self)
def	__del__ (self)

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.1/rocAL_pybind/amd/rocal/plugin/pytorch.py