我有一个预先训练的分割网络，该网络吸收(batch_size,5,224,224)张量并输出一个(batch_size,224,224)特征向量，例如batch_features，最初是Torch张量，但可以转换为numpy数组
有多种收集火炬张量的方法。
选项A：使用列表
val_features = []
val_features.extend(list(batch_features)
***选项B：使用numpy数组*
val_features = np.empty()
val_features = [*val_features, *batch_features]
最后，我想遍历所有这些数组并进行一些后处理。在那种情况下哪种数据格式最合适？