У меня есть предварительно обученная сеть сегментации, которая принимает тензор (batch_size,5,224,224)
и выводит вектор признаков (batch_size,224,224)
, скажем batch_features
, первоначально тензор Факела, но может быть преобразован в массив с нулевыми значениями.
Существует несколько способов сбора тензора горелки.
Вариант A: использовать список
val_features = []
val_features.extend(list(batch_features)
*** Вариант B: использовать массив numy *
val_features = np.empty()
val_features = [*val_features, *batch_features]
В конце я хочу перебрать все эти массивы и выполнить некоторую постобработку. Какой формат данных будет работать лучше в этом случае?