最近想用pytorch搭一个机器翻译模型,过程中涉及到从矩阵中取出某些指定元素组成新矩阵的步骤,然后在pytorch官网上找到对应功能的函数为torch.gather 。pytorch官网上 ...
確定! 回上一頁