torch.gather 作用:收集输入的特定维度指定位置的数值参数: input(tensor): 待操作数。不妨设其维度为(x1, x2, …, xn) dim(int): 待操作的维度。
確定! 回上一頁