torch.where()函数解读
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.7584, 1.1478], [ 0.6773, 0.7763, 0.5024, 0.4952, 0.4198], [ 1.5132, 0.5185, 0.2956, -0.6312, -1.4787]]) >>> b tensor([[1., 1., 1., 1., 1.], [1., 1., 1., 1., 1.], [1., 1., 1., 1., 1.]]) >>> torch.where(a>0,a,b) #合并a,b两个tensor a>0的地方保存a的原值,否则取b的值。 tensor([[1.0000, 1.5895, 1.6003, 1.0000, 1.1478], [0.6773, 0.7763, 0.5024, 0.4952, 0.4198], [1.5132, 0.5185, 0.2956, 1.0000, 1.0000]])
这篇关于torch.where()函数解读的文章就介绍到这儿,希望我们推荐的文章对大家有所帮助,也希望大家多多支持为之网!
- 2024-06-30uniAPP 实现全屏左右滚动滚动的效果-icode9专业技术文章分享
- 2024-06-30如何在本地使用授权或插件-icode9专业技术文章分享
- 2024-06-30伪静态规则配置方法汇总-icode9专业技术文章分享
- 2024-06-29易优CMS安装常见问题汇总-icode9专业技术文章分享
- 2024-06-28易优新手必读安装教程-icode9专业技术文章分享
- 2024-06-28忘记eyoucms后台密码怎么办?-icode9专业技术文章分享
- 2024-06-26终极指南:Scrum中如何设置需求优先级
- 2024-06-26AI大模型企业应用实战(25)-为Langchain Agent添加记忆功能
- 2024-06-26小白家庭 nas 搭建方案-icode9专业技术文章分享
- 2024-06-23AI大模型企业应用实战(14)-langchain的Embedding