torch.where()函数解读

最新推荐文章于 2024-06-02 10:39:13 发布

rrr2

最新推荐文章于 2024-06-02 10:39:13 发布

阅读量797

点赞数

分类专栏： torch 文章标签： pytorch 深度学习机器学习

本文链接：https://blog.csdn.net/qq_35608277/article/details/121694670

版权

torch 专栏收录该内容

49 篇文章 1 订阅

订阅专栏

该博客介绍了Python深度学习库PyTorch中的torch.where()函数。通过示例展示了如何根据条件合并两个tensor，当条件满足时选取其中一个tensor的值，否则选取另一个tensor的值。在例子中，当a中的元素大于0时，选择a的对应元素，否则选择b的元素。

摘要由CSDN通过智能技术生成

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的地方保存
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]])