Нет, я не сталкивался с такими библиотеками, но мы можем написать генератор, который делает это для пустого массива.
def gen(batch):
j = 0
while True:
x = np.array(x_shape)
y = np.array(y_shape)
for i in range(batch_size):
x = read(batch[j])
y = read(batch[j])
j += 1
yeild x,y
Мы можем использовать это как шаблон, мы используем генераторы, потому что наши данные нене вписывается в ОЗУ.мы также можем использовать итератор вместо j при использовании мультиобработки.