python - 如何在不使用后端函数的情况下在keras中编写自定义函数？

Question

我想获取 keras 层的输出(图像张量)，将其转换为 numpy 数组并在不使用后端函数的情况下对其应用自定义函数，然后将输出返回到模型。所以它会是这样的:

def sum(args):
    # convert args to numpy arrays
    ???
    #calculating sum of all the elements
    s=np.sum(args)
    return s

非常感谢您的帮助。

Answer 1

您可以在 Tensorflow 上使用 Keras 使用以下格式:

  x = layers.Conv2D(32, (3, 3), strides=(1, 1), 
    padding='same', name='conv_1', use_bias=False)(input_image)
  x = my_function(x)
  x = layers.BatchNormalization(name='norm_1')(x)