The problem or idea, whatever, is: i want to put my own kernel/filer to conv2d. For example, i have generator of kernels and i want to apply it to batch of images. In tensorflow there are tf.nn.conv2d which have “filter” as input, but these “filters” are of size [in, out, height, width]. No batch size. But i need that. Is there anything like that in pytorch?
Tried to look for answer for this problem too long and found only solution is to make my own conv layer but i hoped that there are already such a thing in pytorch. Or am i wrong?
Thanks in advance