You are viewing a plain text version of this content. The canonical link for it is here.

Posted to commits@tvm.apache.org by GitBox <gi...@apache.org> on 2021/09/24 17:37:45 UTC

[GitHub] [tvm] shashwat14 commented on issue #9078: [Bug] Softmax in TFLite converter is channel first instead of channel last

shashwat14 commented on issue #9078:
URL: https://github.com/apache/tvm/issues/9078#issuecomment-926806805


   I think the axis is hard coded to 1 assuming the task is a simple classification task. However, the softmax can occur over any axis. In the above example it occurs for every "pixel" where each pixel has 14 dimensions. The axis parameter should always  represent the channel index (unless overridden by the user). In numpy notation, axis can be -1 for most cases. 


-- 
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.

To unsubscribe, e-mail: commits-unsubscribe@tvm.apache.org

For queries about this service, please contact Infrastructure at:
users@infra.apache.org