Since skimage uses numpy under the hood, if I’m not mistaken, you would have to use a custom autograd.Function and implement the backward method manually.
Autograd won’t be able to create the backward pass automatically.
I’m not sure which operations are applied for the mentioned operations, but you might be able to implement them using PyTorch operations. If that’s the case, you wouldn’t need to write the backward manually.