RuntimeError:cuda runtime error (11) : invalid argument at /pytorch/aten/src/THC/generic

RuntimeError:cuda runtime error (11) : invalid argument at /pytorch/aten/src/THC/generic/THCTensorMathPointwise.cu:324

在别人的pytorch代碼中，加入自己的子產品，出現的cuda運作時間錯誤。檢視了許多部落客的解釋。弄了好久都沒成功。

這個錯誤主要是加入的子產品中的函數沒有辦法在gpu下執行，将傳入的資料轉成cpu模式，在執行後再調用時再轉成cuda模式運作