网站首页 站内搜索

搜索结果

查询Tags标签: 0.1815,共有 1条记录
  • torch.gather与torch.scatter

    前言 我觉得pytorch最难的两个api之一就是gather和scatter了,但是在项目中又经常出现,OMG… 还是想办法弄懂吧torch.gather input = torch.randn((2, 3, 4))tensor([[[ 0.1815, -0.7603, 0.3465, 0.2593],[ 0.5841, 1.2581, 0.1583, -0.7283],[-0.0093, 0.6214, -0…

    2022/1/23 6:07:52 人评论 次浏览
扫一扫关注最新编程教程