qq_33892106的博客

torch.einsum >>> a = torch.arange(60.).reshape(5,3,4) >>> b = torch.arange(24.).reshape(3,4,2) >>> o = torch.einsum('fnd,ndh->fh', a, b) >>> o tensor([[1012., 1078.], [2596., 2806.], [4180., 4534.],

pytorch 写的好好的代码运行到某一块突然显示 a leaf Variable that requires grad has been used in an in-place operation 这个bug的意思是 需要梯度的叶子变量被用在了原位操作里 指的是 如果是原位操作，就会覆盖原来的值。 因为并不是计算过程，是赋值，梯度是多少这个没有一个定义（或许是一个跳变过程，梯度无穷大）。 所以...