Hi, in your official document of drtk.edge_grad_estimator, there was a note saying masking must not be done before applying edge_grad_estimator. But the example right under it does exactly that. Can ...