**torch.where(conditions,x,y)**可以传递梯度
**tensor.where(conditions)**不能传递梯度,是非差分函数
torch.where能否传递梯度
最新推荐文章于 2024-05-14 10:21:54 发布
**torch.where(conditions,x,y)**可以传递梯度
**tensor.where(conditions)**不能传递梯度,是非差分函数