I couldn’t reproduce. I just installed the latest version from master (which should be virtually equivalent to v0.2.0 for Convolution). I did had to fix an indentation problem with your function though (you need to indent forward to that it belongs to Net).