關於Pytorch的二維tensor的gather和scatter_操作用法分析

看得不明不白（我在下一篇中寫了如何理解gather的用法） gather是一個比較復雜的操作，對一個2維tensor，輸出的每個元素如下： 二維tensor的gather操作 針對0軸 ...