[GitHub] [incubator-mxnet] ChaiBapchya commented on issue #17444: [Large Tensor] Add LT support for NN optimizers and 1 activation function

[GitBox]

ChaiBapchya commented on issue #17444: [Large Tensor] Add LT support for NN 
optimizers and 1 activation function
URL: https://github.com/apache/incubator-mxnet/pull/17444#issuecomment-581680197
 
 
   @szhengac 
   
   Thanks for the help with passing sparse array
   So this works as we discussed offline
   ```
   import mxnet as mx
   from mxnet.test_utils import *
   w = rand_ndarray((2**32+1,1), 'row_sparse', density=1)
   g = rand_ndarray((2**32+1,1), 'row_sparse', density=1)
   m = mx.nd.zeros((2**32+1,1), stype='row_sparse')
   v = mx.nd.zeros((2**32+1,1), stype='row_sparse')
   ans=mx.nd.adam_update(w,g,m,v,lr=0.1)
   ans.data.asnumpy()
   ```
   Output
   ```
   array([[ 0.19461787],
          [-0.3031752 ],
          [-0.18570909],
          ...,
          [        nan],
          [        nan],
          [        nan]], dtype=float32)
   ```

----------------------------------------------------------------
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