我只是使用TensorFlow来实现CNN模型。在这里，我有一个具有4个维度的张量：A = [16,96,96,64]（16是批处理大小，96是图像块大小，而64是特征图的数量）。我只需要计算张量A中每个特征图的平均值（如平均池），然后返回一个新的张量[[16,1,1,64]]，它包含A中每个特征图的平均值
您能告诉我我是怎么实现的吗？
谢谢！
一个张量为[16,96,96,64]的张量，我需要获取该张量中每个特征图的平均值，并返回一个张量为[16,1,1,64]的张量。我不确定如何实现这一目标。
结果应为张量，包括张量A中特征图的平均值，并且其大小为[16,1,1,64]。
答案 0 :(得分：1)
使用带有指定参数tf.reduce_mean()的{{1}}来计算平均值，然后使用axis来还原缩小的轴：
tf.expand_dims()