Method __init__

dali/test/python/operator_2/test_reduce.py:46–57

(self, data_type)

class Batch1D(Batch):
    def __init__(self, data_type):
        super().__init__(data_type)
        self._data = [
            [
                np.array([1, 2, 3, 4], dtype=self._data_type),
                np.array([33, 2, 10, 10], dtype=self._data_type),
            ],
            [
                np.array([10, 20, 30, 20], dtype=self._data_type),
                np.array([33, 2, 15, 19], dtype=self._data_type),
            ],
        ]

    def valid_axes(self):
        return [None, (), 0]