本文分类:news发布日期:2026/3/7 12:34:31
打赏

相关文章

Pytorch随笔:gather函数

gather函数使用方法: gather(dim,index) dim = 0时,使用index填充行;dim = 1时,使用index填充列。 例如,index = tensor([[x,y,z]]) gather(dim=0,index)示意图为: tensor([[(x,0),(y,1),(z,2)]]) gather(dim=1,…

手机版浏览

扫一扫体验

微信公众账号

微信扫一扫加关注

返回
顶部