SyMPC
SyMPC copied to clipboard
Add gradient functions
Description
Add the gradient functions for the list of operations:
- [x] add - PR
- [x] sub - PR
- [x] mul - PR
- [x] sum - PR
- [x] matmul - PR
- [x] reshape - PR
- [ ] mean
- [x] transpose - PR
- [x] sigmoid - PR
- [x] conv2d - PR
- [ ] mse_loss grad
- To be added more
Why?
This is needed for being able to use training.
Additional Context
Check CrypTen and PySyft 0.2.x versions
Some of them are added in #138
will work on the remaining :)
Conv2D might be a good one. CrypTen has good examples on how those can be implemented.