xinyu-intel commented on issue #11086: Infer shape error with softmaxoutput URL: https://github.com/apache/incubator-mxnet/issues/11086#issuecomment-393192905 Yes, you are right. By the way, is it possible to automatically perform flatten before softmax when the softmax input is such (N, C, 1, 1) because some users can identify this dim to be (N, C) and forget to add flatten before that. Thanks!
---------------------------------------------------------------- This is an automated message from the Apache Git Service. To respond to the message, please log on GitHub and use the URL above to go to the specific comment. For queries about this service, please contact Infrastructure at: us...@infra.apache.org With regards, Apache Git Services