utils.py 文件源码

python
阅读 18 收藏 0 点赞 0 评论 0
项目：torchsample 作者: ncullen93 项目源码 文件源码 
def th_gather_nd(x, coords):
    x = x.contiguous()
    inds = coords.mv(th.LongTensor(x.stride()))
    x_gather = th.index_select(th_flatten(x), 0, inds)
    return x_gather
您未登录，请先登录后再评论
评论列表
文章目录

提
问题

写
面经

写
文章

微信
公众号

扫码关注公众号