Thanks for the reply, could you expand on what you mean by slicing and then stacking a little? I don’t see how those operations result in a desired smoothing.
Reply to edit: Oooh Kornia looks really useful and light actually. Do you know if it is supported in tracking gradients for back propagation? (i.e. will it break the gradient tree?)