/demo_custom_layer

How to derive a custom layer in PyTorch? We need the forward and backward code for it! But do we, now?

Primary LanguageTeXMIT LicenseMIT

Stargazers

No one’s star this repository yet.