Tensor.where()

Returns the elements of x or y, depending on condition.

Usage

from tinygrad.tensor import Tensor
 
a1 = Tensor([1, 2, 3])
a1 = a1.where(a1 +1 , a1)
 
print(a1.numpy())

[2. 3. 4.]

;