Generalize convolution op

facebookresearch / diffkt

A framework for automatic differentiation in Kotlin

MIT License

60 stars 6 forks source link

Open alannnna opened 3 years ago

alannnna commented 3 years ago

Needed for MNIST

gafter commented 3 years ago

We do have a convolution op sufficient for mnist (https://github.com/facebookincubator/diffkt/pull/148). But it doesn't handle gradient shapes other than scalar, or higher derivatives.

gafter commented 2 years ago

Or forward derivatives.