使用Tensorflow可以对数组进行可微分类索引吗? 更具体地说,如果我有一个浮点变量以某种方式转换数组的索引,我可以获得变换数组相对于变量的渐变吗? 基于空间变换器网络(https://arxiv.org/pdf/1506.02025v3.pdf和https://github.com/tensorflow/models/blob/master/transformer/spatial_transformer.py)的梯度推导,似乎这应该是可微分的。 我试过实现这个,我遇到了问题,因为我必须在使用tf.gather转换数组之前将转换后的索引转换为整数,并且看起来似乎渐变不能通过它。 有人会建议如何做到这一点吗?