dcslin commented on issue #540: added softmax with axis
URL: https://github.com/apache/incubator-singa/pull/540#issuecomment-538955169
 
 
   > still has some problems, the output of multiple dimension inputs is not 
correct.
   > please check:
   > 
   > ```
   > x_0 = np.array([[0, 1, 2, 3], [10000, 10001, 10002, 
10003]]).astype(np.float32)
   > # axis is 1
   > # expected output [[0.0320586, 0.08714432, 0.23688284, 0.64391428],
   > #                 [0.0320586, 0.08714432, 0.23688284, 0.64391428]]
   > ```
   
   Hi @joddiy , this is updated, could you please help to review?

----------------------------------------------------------------
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.
 
For queries about this service, please contact Infrastructure at:
us...@infra.apache.org


With regards,
Apache Git Services

Reply via email to