Issue
I only found the conv2d implement by torch.nn.functional.conv2d,and implemented by C.but I counld found which cpp implement it.enter image description here
I try to find the cpp which implement the conv2d in the pytorch files(find how to caculate the conv2d)
Solution
You can find the implementation of conv2d
in the main branch of the pytorch github repository.
The variations of the function conv2d and the convolution itself are defined here
Answered By - rot8
0 comments:
Post a Comment
Note: Only a member of this blog may post a comment.