【问题标题】:How to take an average /sum of (5D tensor) sequence of images? becomes 4D tensor如何获取(5D张量)图像序列的平均/总和?变成 4D 张量
【发布时间】: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


【解决方案1】:

使用reduce_mean():

tf.math.reduce_mean(input_tensor, axis=-1)

【讨论】:

    猜你喜欢
    • 1970-01-01
    • 1970-01-01
    • 2021-07-19
    • 1970-01-01
    • 2022-01-18
    • 1970-01-01
    • 1970-01-01
    • 2021-02-19
    • 1970-01-01
    相关资源
    最近更新 更多