标签： python tensorflow
我想用2元D张量中的每个元素x替换该元素f（x）的函数。
对于1-D张量，tf.map_fn（）可以完成这项工作。我为高阶张量寻求类似的解决方案。
答案 0 :(得分：0)
简单地展平并重塑：
tf.reshape(tf.map_fn(f, tf.reshape(a, [-1])), a.shape)