Hi,
You can access this from the backward method that gets the same ctx as input.
I don’t think you can extract it in a reliable way though.
Why do you need to do this? Can’t you just return it as another output of the forward method?