torch.where能否传递梯度

**torch.where(conditions,x,y)**可以传递梯度
**tensor.where(conditions)**不能传递梯度,是非差分函数

全部评论

相关推荐

流浪的神仙:无恶意,算法一般好像都得9硕才能干算法太卷啦
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务