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