如何用f（x）代替二维张量的每个元素x？

Question

我想用元素f（x）的函数替换二维张量中的每个元素x。你知道吗

对于一维张量，地图（）可以做这项工作。对于高阶张量，我寻求一个类似的解。你知道吗

Answer 1

简单展平和重塑：

tf.reshape(tf.map_fn(f, tf.reshape(a, [-1])), a.shape)