使用 TL 激活的 2 次导数出现问题
作者: seraph0012创建于 2021年10月9日更新于 2021年10月9日
你好! 我在一个模型上工作,需要计算输出与某些输入变量之间的二阶导数。如果我使用 tensorflow 中的现有激活函数(例如 tf.nn.tanh),二阶导数将被正确计算。但是,如果我用 tensorlayer 中的任何激活函数替换 tensorflow 中的激活函数,第一阶导数仍然会被计算,但第二阶导数将为 None。我使用的是 Python 3.8.10、Tensorflow 2.5.0 和 Tensorlayer 2.2.3。以下是重现问题的代码:
内容来源: tensorlayer/TensorLayer