我有一个用tensorflow编写的函数，它操作数组 . 它接收2d数组并对数组的一部分执行一些操作，然后输出相同大小和形状的数组，但是使用修改后的元素 . 我的问题是，在我的模型中，我不是一次只得到一个数组，我得到一批形状（batch_size，array_size，channel） . 由于通道通常是单声道，因此输入批次是4D张量 . 有没有什么方法可以将我的函数应用到这个4D张量中的每一层（矩阵），而不会显着修改函数？
功能如下：
def myFunction(matrix):
# function body
return another_matrix
我怎么能将它应用于一批5个这样的矩阵？
1 回答
tf.map_fn
它将
f应用于每个元素，在您的情况下应用于每个2D数组 . 通过解包第二个参数获得这些2D阵列 .