warp.where#

warp.where(
cond: bool,
value_if_true: Any,
value_if_false: Any,
) → Any#

  • Kernel

  • Differentiable

Select between two arguments, if cond is True then return value_if_true, otherwise return value_if_false.

warp.where(
cond: int8,
value_if_true: Any,
value_if_false: Any,
) → Any

  • Kernel

  • Differentiable

Select between two arguments, if cond is True then return value_if_true, otherwise return value_if_false.

warp.where(
cond: uint8,
value_if_true: Any,
value_if_false: Any,
) → Any

  • Kernel

  • Differentiable

Select between two arguments, if cond is True then return value_if_true, otherwise return value_if_false.

warp.where(
cond: int16,
value_if_true: Any,
value_if_false: Any,
) → Any

  • Kernel

  • Differentiable

Select between two arguments, if cond is True then return value_if_true, otherwise return value_if_false.

warp.where(
cond: uint16,
value_if_true: Any,
value_if_false: Any,
) → Any

  • Kernel

  • Differentiable

Select between two arguments, if cond is True then return value_if_true, otherwise return value_if_false.

warp.where(
cond: int32,
value_if_true: Any,
value_if_false: Any,
) → Any

  • Kernel

  • Differentiable

Select between two arguments, if cond is True then return value_if_true, otherwise return value_if_false.

warp.where(
cond: uint32,
value_if_true: Any,
value_if_false: Any,
) → Any

  • Kernel

  • Differentiable

Select between two arguments, if cond is True then return value_if_true, otherwise return value_if_false.

warp.where(
cond: int64,
value_if_true: Any,
value_if_false: Any,
) → Any

  • Kernel

  • Differentiable

Select between two arguments, if cond is True then return value_if_true, otherwise return value_if_false.

warp.where(
cond: uint64,
value_if_true: Any,
value_if_false: Any,
) → Any

  • Kernel

  • Differentiable

Select between two arguments, if cond is True then return value_if_true, otherwise return value_if_false.

warp.where(
arr: Array[Any],
value_if_true: Any,
value_if_false: Any,
) → Any

  • Kernel

  • Differentiable

Select between two arguments, if arr is not null then return value_if_true, otherwise return value_if_false.