tensorflow - 如何找到函数最大的值（实现代码）？

Question

我正在尝试解决以下问题：我有这个功能：

I0(k) 是这个函数：

.

在我的代码的每一步中，我都有一个已知的、不同的余弦角度，我希望每次都知道：对于哪个 k，函数最大。在数学上，我必须计算导数，将其设置为零并求解方程。但是如何用 tensorflow 来实现呢？请问解决这个问题的最好方法是什么？