Incubator-mxnet: Numpy Diag in MXNet
Borrow from https://discuss.gluon.ai/t/topic/3846.
We can consider to implement np.diag in the future.
https://docs.scipy.org/doc/numpy-1.13.0/reference/generated/numpy.diag.html
sxjscience
👍2
All 4 comments
I am interested implementing this, is there a jira issue already?
ifeherva
on 30 Jun 2018
@ifeherva NoJIRA issue for this one yet.
sxjscience
on 4 Jul 2018
I am working on replicating np.diag(). Once done will do the full diagonal operator. This is my first operator so I am learning a lot at the moment.
- [x] 2d forward pass CPU
- [x] 2d forward pass GPU
- [x] 1d forward pass CPU
- [x] 1d forward pass GPU
- [x] 2d backward pass CPU
- [x] 2d backward pass GPU
- [x] 1d backward pass CPU
- [x] 1d backward pass GPU
- [x] unit tests
ifeherva
on 8 Jul 2018
🎉4
@eric-haibin-lin This issue can be closed now that the PR has been merged
samskalicky
on 31 Jul 2018
Was this page helpful?
0 / 5 - 0 ratings
Related issues
sbodenstein
路
3Comments
phunterlau
路
3Comments
Ajoo
路
3Comments
seongkyun
路
3Comments
realbns2008
路
3Comments
Most helpful comment
I am working on replicating np.diag(). Once done will do the full diagonal operator. This is my first operator so I am learning a lot at the moment.
PR:
https://github.com/apache/incubator-mxnet/pull/11643