e.g.
out = Net(input)
out shape : batchSize * C * H * W
what if I want to do some convolution operation(like Gaussian) to the output ?
The straightforward way is to do that in a loop for each sample. But it is infeasible for the
consideration of efficiency.
Is there any good idea ?