hi,
I’m afraid you won’t be able to use skimage and pytorch’s autograd. To work, the autograd needs definition of the derivatives for everything that happens. So we only support pytorch ops.
If you know how to write the backward for that part though, you can create a custom Function and define the backward yourself. See doc here for more details.