grain
grain copied to clipboard
Capsule Networks
Thanks for your request. I've never read the paper yet but just check out pytorch implentation https://github.com/timomernick/pytorch-capsule There are several missing ops in grain
- binary ops: max
- reductions ops: mean, sum, max
- topology ops: slice, stack, transpose, cat, view (these are relatively harder to implement) To support topology ops, I gonna make huge changes in variable struct. Then the method implementation can be done.
Nice. Thanks for your fast answer.