如何在张量流内核代码中改变变量张量的值？

Question

我正在尝试在计算功能中更改ctx(OpKernelContext)中的输出值。当通过常量执行操作时，我可以通过tensor->flat<>()检查值。

但是，当变量执行操作时，我找不到它们。如果我执行flat<>()，则会返回细分错误。我想在一些具有一些可变输入的操作中访问输出值。

Answer 1

无法重现有关您尝试过的内容的详细信息。

也就是说，直接访问变量张量的内容不应导致段错误。如果您尝试取消对CPU内核中GPU内存的指针的引用，可能会发生这种情况。