I'm trying to implement a complex model, then visualize it with Captum's integrated gradients library (https://captum.ai/docs/extension/integrated_gradients). However, Captum's implementation expects everything to be a torch tensor, not a Cplx object. Any thoughts for a workaround?
I'm trying to implement a complex model, then visualize it with Captum's integrated gradients library (https://captum.ai/docs/extension/integrated_gradients). However, Captum's implementation expects everything to be a torch tensor, not a Cplx object. Any thoughts for a workaround?