tensorflow-lite - TFLite张量android

Question

我使用 TFLite .a 静态库在 android 上集成了 TFLite。TFLite 支持 TfLiteTensor 类，不支持 Tensor 类。如何在 tensorflow lite 上使用 tensor，如 tensorflow 张量操作？例如张量升维、降维和张量切片。谢谢！

标签： tensorflow-lite

Answer 1

（从https://github.com/tensorflow/tensorflow/issues/47138复制）

我不确定我是否正确理解，但如果您想尝试这些操作是否在 TFLite 模型中工作，您可以制作一个 TF 模型并将其转换为 TFLite 模型。TFLite 中的 RESHAPE 和 STRIDED_SLICE 操作可以部分支持维度操作和切片操作。

张量上的操作实际上是计算图中的另一个操作。话虽如此，图表就像

TfLiteTensor -> RESHAPE op -> 具有新形状的新 TfLiteTensor。