Gathers values along an axis specified by dim . For a 3-D tensor the output is specified by: out[i][ ...
確定! 回上一頁