I got this problem when trying to use maskedtensor in backpropagation. But it works well without maskedtensor. It seems like something wrong with maskedtensor. But I don’t know how to fix it.
torch 1.12.0
maskedtensor 0.10.0