Does anyone know of a pytorch differentiable filter that does the same thing?