Python tensorflow中的可微直方图

如何在tensorflow中得到一个可微的直方图函数。我想用图像的直方图来计算一个特定的损失函数。但是tensorflow提供的直方图函数tf.histogram\u fixed\u width（）是不可微的。我怎样才能做到这一点？

一个可微损失可以自然地定义在两个大小相同的张量之间，关于每个张量的直方图（或者更准确地说，argsort）。