PyTorch中的torch.gather函数,可以帮助我们在指定维度上按照索引从输入张量中获取对应的值,并输出一个新的张量。这个函数的操作可能比较抽象,下面我...

╰+攻爆jí腚メ 2024-03-26 20:53 6阅读 0赞

发表评论

表情:
评论列表 (有 0 条评论,6人围观)

还没有评论,来说两句吧...

相关阅读