【发布时间】:2021-01-23 08:54:08
【问题描述】:
我有一个来自特征提取模块的 5D 图像序列(无、4、10、4、1536),但我需要对所有序列特征进行平均/求和,以便适合 2DCNN 层(4D 张量),如何在 TensorFlow 中具体做到这一点?提前致谢。
【问题讨论】:
-
试试
tf.math.reduce_mean(input_tensor, axis=-1) -
谢谢!!还有一个问题,我想尝试将它与另一个 4d 层连接起来,reduce_mean 的结果形状不匹配
Concatenate层需要具有匹配形状的输入,除了 concat 轴。得到输入形状:[(None, 4, 10, 64), (None, 10, 4, 64)] 如何重新排序张量内的值? -
使用
tf.transpose()
标签: python tensorflow transform tensor