Reshaping output tensor

85 views
Skip to first unread message

sai_ng

unread,
Aug 29, 2021, 11:21:04 PM8/29/21
to TensorFlow Lite
Hey all,

I'm working on adding image-to-image support for models in tensorflow-lite. For this to work, I want the model input/output tensor to be dynamic shape (dims_signature -1).

I could re-size input tensor using "ResizeInputTensor()" method, but I'm unable to resize output tensor. For more info: Check this pull request.



Best
NGS
Reply all
Reply to author
Forward
0 new messages