Input to IdentityAttachKLSparseReg Symbol

Hi ,

I am trying to use the IdentityAttachKLSparseReg symbol for one of my use cases but I couldnt get it working. Based on the documentation here ,

am providing a Single Input argument ‘data’ to the symbol. Here’s my code along with the error :

import mxnet as mx
data = mx.symbol.Variable(“data”)
data_nd = mx.nd.array([1,1,1,1])
out = mx.symbol.IdentityAttachKLSparseReg(data)
exe = out.bind(mx.cpu(), {‘data’ : data_nd })
Traceback (most recent call last):
File “”, line 1, in
File “/usr/local/lib/python2.7/dist-packages/mxnet/symbol.py”, line 1607, in bind
‘aux_states’, aux_states, self.list_auxiliary_states(), False)
File “/usr/local/lib/python2.7/dist-packages/mxnet/symbol.py”, line 1242, in _get_ndarray_inputs
raise ValueError(‘Length of %s does not match the number of arguments’ % arg_key)
ValueError: Length of aux_states does not match the number of arguments

Can someone point me where I am wrong ? or the right way to use the symbol ?

TIA.
Psn9

It seems that its a bug from Mxnet.

Will post a solution when the developers fix it.