标签： c++ tensorflow
我正在为TensorFlow中的GPU实现编写自定义操作，我想知道是否有一个标准库可供人们使用琐碎的内核。我正在实现的内核，类似于pytorch EmbeddingBag的Op的向后传递，有一种简单的情况，我想返回零梯度。在TensorFlow回购中，我找到了其他内核的函子，但它们不是pip包的一部分，因此我猜它们不应该被重复使用。例如，在PyTorch中，它们具有ATen库，但是TensorFlow缺少类似的东西，对吧？