Is there a function like torch.nn.fold in JAX? #25352
Unanswered
ZhongkuiMa
asked this question in
Q&A
Replies: 0 comments
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
-
Hi Everyone,
I have known that conv_general_dilated_patches can realize a function like the torch.nn.unfold.
But I need an implementation like the torch.nn.fold.
Is there a function in JAX that can achieve this?
Or should I implement it by myself? : )
Thank you
Beta Was this translation helpful? Give feedback.
All reactions