Hi, all
My code like this:
output=torch.bmm(l_atten.unsqueeze(1),l_feature)
return output
where l_atten is obtained from l_feature according to a NNs
So both l_feature and l_atten need backward.
1 Like
Can you format your code and text properly? If you do not want to show your effort in trying to solve your problem, no one will. If you do not know how to use Markdown, see tutorial here.