silvester:

There is no error without F.softmax .


This would mean that F.softmax needs its output to calculate the gradients (as also given in the definition in derivatives.yaml 1). This would disallow all inplace manipulations in the model output.