标签: tensorflow tensorflow-lite
我正在编写一个自定义op,该op输出一个张量,其形状取决于输入张量的值。问题在于我们无法访问Prepare方法中的张量值。我们可以得到张量 shapes ,但是 values 不可用。我该如何实现?
Prepare
在相关说明中,我如何支持输出具有部分指定形状的张量?需要在eval函数期间分配张量,但是我看不到在运行时分配张量的API。
eval