在tensorflow的自定义层中使用numpy的外部函数

时间:2019-12-18 21:44:16

标签: numpy tensorflow tf.keras

我在tensorflow 2.0中有一个带有一个自定义层的简单模型。 (我在自定义图层上关注this tensorflow.org tutorial

call()类的MyDenseLayer方法中,我想使用通过使用numpy编写的外部函数。这有可能吗?

如果仍然无法正常工作,如果我将所有numpy用作tf.function(外部函数),是否可以正常工作?

0 个答案:

没有答案