前面已经介绍了index_select 和mask_select 两个选择函数,这两个函数通过一定的索引规则从输入张量中筛选出满足条件的元素值,只不过in... torch.nonzero.
確定! 回上一頁