Question

我目前正在关注a tutorial in Pytorch，下面是这个表达式：

grad_h[h < 0] = 0

此语法如何工作？它有什么作用？

Answer 1

这意味着将grad_h中对应的h为负的所有值替换为零。

因此，它正在实现某种掩码，以仅在h为负值时保持渐变值

假设grad_h和h具有相同的形状。

grad_h.shape == h.shape

当您h < 0进行操作时，如果每个i为True，则将获得一组形状相同的布尔值数组，并将它们设置为h[i] < 0。

因此，您可以应用此蒙版对grad_h进行切片，最后将所有切片的元素设置为零

Answer 2

这意味着只要h小于0，变量grad_h就等于0。