PyTorch 入门笔记-gather选择函数 · input (Tensor) - 输入张量 · dim (int) - 需要进行索引的轴 · index (LongTensor) - 要采集元素的索引 · sparse_grad ( ...
確定! 回上一頁