tarch.gather(input,dim,index)执行查表操作,input是一个表格,在dim维度中查询index内容
torch.gather
最新推荐文章于 2024-05-29 01:07:50 发布
tarch.gather(input,dim,index)执行查表操作,input是一个表格,在dim维度中查询index内容