Module . forward function here means the forward function of the torch. ... Both the grad_inputs are size [5] but shouldn't the weight matrix of the linear ...
確定! 回上一頁