You can use pytensor.tensor.where
to obtain the same behavior. Would be good to add a helper for that in PyTensor
1 Like
You can use pytensor.tensor.where
to obtain the same behavior. Would be good to add a helper for that in PyTensor