Extend to support Conv1d and Conv3d

cybertronai / autograd-hacks

The Unlicense

151 stars 31 forks source link

Extend to support Conv1d and Conv3d #12

Closed swilson314 closed 3 years ago

swilson314 commented 3 years ago

I suspect supporting Conv1d and Conv3d is straight forward, but einsum() probably needs to change? Does anyone know the correct values here for the computation string?

swilson314 commented 3 years ago

Extension to Conv1d/3d is straightforward copy of Conv2d code, but unfold() that supports 3/4/5D tensors is required. The following worked https://github.com/f-dangel/unfoldNd