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.