input(Tensor) - 需要进行索引操作的输入张量;; dim(int) - 需要对输入张量进行索引的维度;; index(LongTensor) - 包含索引号的1D 张量;; out(Tensor, ...
確定! 回上一頁