Function empty_batches

dali/test/python/operator_2/test_reduce.py:603–614

(rank, axes, batch_size, num_batches)

def empty_batches(rank, axes, batch_size, num_batches):
    data = []
    for _ in range(num_batches):
        batch = []
        for _ in range(batch_size):
            shape = np.random.randint(1, 10, size=rank)
            for a in axes:
                shape[a] = 0
            sample = np.empty(shape, dtype=np.float32)
            batch.append(sample)
        data.append(batch)
    return data


@nottest