本文將介紹在使用深度學習框架pytorch中的兩個常用的函數,這兩個函數在搭建模型的時候會經常用到, ... torch.gather(input,dim,index,out=None).
確定! 回上一頁