torch expand dim gather(input, dim, index, out=None, For example, if input is 4x10x15 and dim=0 than index has to be Nx10x15. Align axes for broadcasting.
確定! 回上一頁