张量流:可以区分索引吗?

时间:2016-07-28 00:20:26

标签: tensorflow

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

0 个答案:

没有答案