【pytorch】RuntimeError: Integer division of tensors using div or / is no longer supported【解決】

ok， 能搜到這篇文章大概遇到了我已經遇到過的問題。今天把pytorch升級到1.6.0，發現tensor和int之間的除法不能直接用'/'。明明1.5.0都是能夠用的-_-。火炬這種鄰代兼容性有點值得吐槽。

對於這個問題直接看官方文檔就能夠了：

https://pytorch.org/docs/stable/generated/torch.div.html

或者，看個人解決方案：

對於tensor A和整數n之間的除法：

result = A / n # not supported in torch 1.6.0


# solution
result = torch.floor_divide(A, n)

這個floor_divide至關於python中的'//'，即獲得的結果爲整型(去掉了小數點後的數字)。

若是你不想要這種除法，想獲得帶小數點的準確數值，您能夠：

result = torch.true_divide(A, n)

根據具體狀況選取以上兩種除法，便可解決這個issue。