网站首页 站内搜索

搜索结果

查询Tags标签: 1.1478,共有 2条记录
  • torch.where()函数解读

    函数作用 torch.where()函数可以是按照给定的条件合并两个tensor; 条件满足的位置取前者,否则取后者。 代码示例>>> import torch >>> a=torch.randn(3,5) >>> b=torch.ones(3,5) >>> a tensor([[-0.0310, 1.5895, 1.6003, -1.75…

    2021/12/3 23:21:21 人评论 次浏览
  • torch.where()函数解读

    函数作用 torch.where()函数可以是按照给定的条件合并两个tensor; 条件满足的位置取前者,否则取后者。 代码示例>>> import torch >>> a=torch.randn(3,5) >>> b=torch.ones(3,5) >>> a tensor([[-0.0310, 1.5895, 1.6003, -1.75…

    2021/12/3 23:21:21 人评论 次浏览
扫一扫关注最新编程教程