Лучший выходной формат для хранения результатов, собранных после партии - PullRequest
0 голосов
/ 18 марта 2019

У меня есть предварительно обученная сеть сегментации, которая принимает тензор (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]

В конце я хочу перебрать все эти массивы и выполнить некоторую постобработку. Какой формат данных будет работать лучше в этом случае?

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...