Tensor-Puzzles
Tensor-Puzzles copied to clipboard
Problem with `where` function
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])