在tensorflow2中,用numpy运算函数构建loss函数没有梯度的问题

在tensorflow2中,如果单纯用numpy运算函数构建loss函数会报错无法求梯度,但如果在loss中加上一个tf自身的可以求梯度的部分,例如加上crossentropy后,便可以进行训练,只是我不太清楚在这个时候,用numpy写成的那部分究竟有没有生效

不知道你这个问题是否已经解决, 如果还没有解决的话:

如果你已经解决了该问题, 非常希望你能够分享一下解决方案, 以帮助更多的人 ^-^