pytorch:数学运算和近似值的clamp操作

阅读量：

复制代码

    import torch

±*/

当所涉及的维度存在差异时，系统将执行广播操作

复制代码

    a = torch.ones(2,2)
    b = torch.tensor(4)
    print(a + b) #用到了broadcast
    print(a.add(b))
    print(torch.add(a,b))

复制代码

    tensor([[5., 5.],
        [5., 5.]])
    tensor([[5., 5.],
        [5., 5.]])
    tensor([[5., 5.],
        [5., 5.]])

复制代码

    print(a/b)
    print(torch.all(torch.eq(a/b,a.div(b))))
    print(torch.equal(a/b,a.div(b))) 
    #ps:torch.eq(a,b) 逐一比较a,b中

还没有任何评论哟~