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?
And why is oshape fixed at 2 dimensions? cant it be higher dimensions now with your improvements? [ Full content available at: https://github.com/apache/incubator-mxnet/pull/12430 ] This message was relayed via gitbox.apache.org for [email protected]
