torch.gather creates a new tensor from the input tensor by taking the values ... but the number of columns will equal the length of each list in index .
確定! 回上一頁