如何在张量流内核代码中更改变量类型张量中的值

How to change value in variable type tensor in tensorflow kernel code?

本文关键字：变量 改变 类型 张量中 张量流 内核 代码      更新时间：2023-10-16

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

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

这不可能通过有关您尝试过的内容的更多详细信息来重现。

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

相关文章：