官方說明 gather可以對一個Tensor進行聚合,宣告為:torch.gather(input, dim, index, out=None) → Tensor 一般來說有三個引數:輸入的變數input.
確定! 回上一頁