shashwat14 removed a comment 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: [email protected] For queries about this service, please contact Infrastructure at: [email protected]
