grain icon indicating copy to clipboard operation
grain copied to clipboard

Capsule Networks

Open nordlow opened this issue 6 years ago • 2 comments

Are you planning on adding support for capsule networks?

:)

nordlow avatar Jun 27 '18 13:06 nordlow

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.

ShigekiKarita avatar Jun 28 '18 01:06 ShigekiKarita

Nice. Thanks for your fast answer.

nordlow avatar Jun 28 '18 15:06 nordlow