Pytorch高阶OP操作where,gather原理
更新时间:2020年04月30日 11:16:00 作者:杰鑫哥
这篇文章主要介绍了Pytorch高阶OP操作where,gather原理,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下
PyTorch是一个非常有可能改变深度学习领域前景的Python库。我尝试使用了几星期PyTorch,然后被它的易用性所震惊,在我使用过的各种深度学习库中,PyTorch是最灵活、最容易掌握的。
一、where
1)torch.where(condition, x, y) # condition是条件,满足条件就返回x,不满足就返回y
2)特点,相比for循环的优点是:可以布置在GPU上运行
二、gather
1)官方解释:根据指定的维度和索引值来筛选值
2)举例
以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持脚本之家。
相关文章
Tensorflow中k.gradients()和tf.stop_gradient()用法说明
这篇文章主要介绍了Tensorflow中k.gradients()和tf.stop_gradient()用法说明,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧2020-06-06pandas dataframe获取所有行名称与列名称方法示例
这篇文章主要给大家介绍了关于pandas dataframe获取所有行名称与列名称的相关资料,Pandas是Python中用于数据分析的非常重要的库,它提供了多种方法来获取列名,需要的朋友可以参考下2023-09-09
最新评论