Yes, you are right. Do you have any suggestion? I don’t know where to start. Should I check torch.jit.trace to apply kernel to F.conv3d? or just make custom C++/CUDA extended convolution to avoid problem of tracing?