torch:sub

(torch:sub a b)

numpy スタイルのブロードキャストを伴う、微分可能な要素ごとの a - b (linalg:sub) です。第 2 オペランドの勾配は符号が反転します (ブロードキャストの縮約は torch:add と同じです)。