实现带梯度矩阵指数的新tensorflow GPU op的最佳方法

Question

我正在使用一个大型框架，希望从经验丰富的程序员那里获得一些建议。

这是我的计划：

• 使用Eigen不支持的MatrixFunction或已经存在的tensorflow矩阵指数op来为矩阵指数实现新的tensorflow GPU op
• 添加渐变
• 使用我的c ++库中的新操作，该库已经使用tensorflow作为第三方。

我知道tensorflow具有矩阵指数运算，但据我所知它不适用于GPU，也没有渐变实现。我是否应该将此功能添加到现有操作中，而不是注册一个新功能？以及在用户操作中使用Eigen不支持的MatrixFunction怎么办？