if the input shape is (3,3) then the output diagonal will be (3), will this cause a problem with indexing j[1] since there wont be a 2nd dimension in the oshape?
[ Full content available at: https://github.com/apache/incubator-mxnet/pull/12430 ] This message was relayed via gitbox.apache.org for [email protected]
