pytorch pow_()使用需谨慎

示例:

    x = torch.randn(1)
    y1 =x
    print("x:",x)
    print("y1:",y1)

    y2 = x.pow_(2)
    print("pow_ ...")
    print("x:",x)
    print("y1:",y1)
    print("y2:",y2)

输出:

x: tensor([-1.7124])
y1: tensor([-1.7124])
pow_ ...
x: tensor([2.9323])
y1: tensor([2.9323])
y2: tensor([2.9323])

结论:
1.pow_()进行了原地操作,由于传递的是引用,x.pow_(2)后,x和y1的值都改变了
2.使用x**2或torch.pow(x,2)代替,不进行原地操作,x和y1不改变

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值