今天写代码的时候遇到一个问题，网络前向过程中有一个张量A，我想把张量A中的大于0的值变成张量B中对应的值，最初的实现是：
A[A>0]=B[A>0]
然后运行起来就报错了，原因是这个操作属于in-place操作，而pytorch在涉及到求梯度的tensor时，是不允许对这些tensor做原地操作的，否则在反向传播的时候，这些张量计算出来的梯度发生变化。
所以我后来采用了torch.where()方法：
torch.where(condition, x, y) → Tensor
# 使用where方法
C = torch.where(A > 0, B, A)
condition为y的条件表达式，where方法检查y中的所有元素，对于y中满足condition的元素，用x中对应元素替换；否则，还保留y中的元素。where返回一个新的张量。
今天写代码的时候遇到一个问题，网络前向过程中有一个张量A，我想把张量A中的大于0的值变成张量B中对应的值，最初的实...
之前对于这一方面了解较少，搭网络就是直接一层一层堆砌，简单粗暴。当然这样做是不对的，写出来的代码不仅难看，集成度不...
目录： PyTorch学习教程、手册 PyTorch视频教程 NLP&PyTorch实战 CV&PyTorch实战...
1.深度学习框架 Pytorch https://github.com/pytorch/pytorch PyTor...
1.深度学习框架 Pytorch https://github.com/pytorch/pytorch PyTor...
1.深度学习框架 Pytorch https://github.com/pytorch/pytorch PyTor...
PyTorch学习资源整理： 最先看，PyTorch官方60min入门指南 对照着看，PyTorch深度学习：60...
去掉大小为1的维度 x = torch.squeeze(x) 添加大小为1的维度 x = torch.unsque...
在跑CIN的代码时,将batch_size从10一路降到2,依然每执行sample就爆显存.请教师兄后,发现问题出...
pytorch官网 https://pytorch.org/ 根据官网提示安装，选择pip方式安装即可（因为con...
本文标题：pytorch学习经验（六）torch.where()：根据条件
本文链接：https://www.haomeiwen.com/subject/wtcfzktx.html
网友评论