Problem with `where` function

[zarif98sjs]

Shouldn't the where function be this?

def where(q, a, b):
    "Use this function to replace an if-statement."
    return (q * a) + (torch.logical_not(q)) * b

Otherwise if we use ~q, technically isn't that incorrect according to the desired function outcome?

If we used ~q, where(arange(4) * 0, 0, 1) returns tensor([-1, -1, -1, -1]). But the desired output should be tensor([1, 1, 1, 1])