zoukankan      html  css  js  c++  java
  • torch.gt(a,b)----->pointwise元素比较

    """
    两个同shape的tensor,point-wise级别元素对应比较,前者大于后者,返回位置元素为1;前者小于等于后者,返回位置元素为0
    """
    >>> import torch
    >>> a=torch.randn(2,4)
    >>> a
    tensor([[-0.5466,  0.9203, -1.3220, -0.7948],
            [ 2.0300,  1.3090, -0.5527, -0.1326]])
    >>> b=torch.randn(2,4)
    >>> b
    tensor([[-0.0160, -0.3129, -1.0287,  0.5962],
            [ 0.3191,  0.7988,  1.4888, -0.3341]])
    >>> torch.gt(a,b)            #得到a中比b中元素大的位置,由于广播作用, b也可以为1个数!!!
    tensor([[0, 1, 0, 0],
            [1, 1, 0, 1]], dtype=torch.uint8)
    >>> torch.gt(b,a)           #b中比a中大
    tensor([[1, 0, 1, 1],
            [0, 0, 1, 0]], dtype=torch.uint8)
    >>> torch.gt(a,1)
    tensor([[0, 0, 0, 0],
            [1, 1, 0, 0]], dtype=torch.uint8)
    >>> c=torch.Tensor([[1,2,3],[4,5,6]])
    >>> d=torch.Tensor([[1,1,3],[5,5,5]])
    >>> torch.gt(c,d)                   #必须是严格大于才为1
    tensor([[0, 1, 0],
            [0, 0, 1]], dtype=torch.uint8)
    
    
    
  • 相关阅读:
    自动生成四则运算题目
    学习进度总结随笔
    作业1
    软件工程项目总结
    结对编程项目---四则运算
    作业三
    自动生成小学四则运算题目的程序
    学习进度总结
    大三下自我简介
    寒假社会实*报告
  • 原文地址:https://www.cnblogs.com/Henry-ZHAO/p/13857435.html
Copyright © 2011-2022 走看看