我想获取 keras 层的输出(图像张量),将其转换为 numpy 数组并在不使用后端函数的情况下对其应用自定义函数,然后将输出返回到模型。所以它会是这样的:
def sum(args):
# convert args to numpy arrays
???
#calculating sum of all the elements
s=np.sum(args)
return s
非常感谢您的帮助。
最佳答案
您可以在 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)
关于python - 如何在不使用后端函数的情况下在keras中编写自定义函数?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/52106943/