标签: numpy tensorflow tf.keras
我在tensorflow 2.0中有一个带有一个自定义层的简单模型。 (我在自定义图层上关注this tensorflow.org tutorial。
在call()类的MyDenseLayer方法中,我想使用通过使用numpy编写的外部函数。这有可能吗?
call()
MyDenseLayer
如果仍然无法正常工作,如果我将所有numpy用作tf.function(外部函数),是否可以正常工作?
tf.function