欢迎来到代码驿站!

Python代码

当前位置:首页 > 软件编程 > Python代码

Pytorch高阶OP操作where,gather原理

时间:2020-12-10 09:33:08|栏目:Python代码|点击:

PyTorch是一个非常有可能改变深度学习领域前景的Python库。我尝试使用了几星期PyTorch,然后被它的易用性所震惊,在我使用过的各种深度学习库中,PyTorch是最灵活、最容易掌握的。

一、where

1)torch.where(condition, x, y) # condition是条件,满足条件就返回x,不满足就返回y

2)特点,相比for循环的优点是:可以布置在GPU上运行

二、gather

1)官方解释:根据指定的维度和索引值来筛选值

2)举例

上一篇:目前最全的python的就业方向

栏    目:Python代码

下一篇:下载糗事百科的内容_python版

本文标题:Pytorch高阶OP操作where,gather原理

本文地址:http://www.codeinn.net/misctech/31474.html

推荐教程

广告投放 | 联系我们 | 版权申明

重要申明:本站所有的文章、图片、评论等,均由网友发表或上传并维护或收集自网络,属个人行为,与本站立场无关。

如果侵犯了您的权利,请与我们联系,我们将在24小时内进行处理、任何非本站因素导致的法律后果,本站均不负任何责任。

联系QQ:914707363 | 邮箱:codeinn#126.com(#换成@)

Copyright © 2020 代码驿站 版权所有