Generalised Conv1x1 to n dimensions

didriknielsen / survae_flows

Code for paper "SurVAE Flows: Surjections to Bridge the Gap between VAEs and Flows"

MIT License

283 stars 34 forks source link

Closed Nintorac closed 3 years ago

Nintorac commented 3 years ago

Allows Conv1x1 to be used with 1, 2, and 3d tensors

Nintorac commented 3 years ago

No problem. How does that look?