У меня есть список списков "all_lists", который содержит 10 списков;
all_lists = [list0, list1, list2, list3, list4, list5, list6, list7, list8, list9]
теперь, когда классификатор предсказывает метку как 3, я помещаю ее в all_list [index = 3 ] так (например, list3 в данном случае),
preds = classifier(xs)
_, predicted = torch.max(preds.data, 1)
index1 = predicted.item()
if predicted == ys:
all_lists[index1].append((xs,ben))
, но это работает только для batch_size = 1, как я могу сделать это для batch_size> 1, чтобы данные добавлялись в соответствующий список?