torch.index_select函数,顾名思义就是根据index索引在input输入张量中选择某些特定的元素,下面介绍该函数的参数。torch.index_select(input, dim, ...
確定! 回上一頁