warp.select#

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

  • Kernel

  • Differentiable

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

Removed in version 1.10: Use where() instead, which has the more intuitive argument order: where(cond, value_if_true, value_if_false).

Deprecated since version 1.7.

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

  • Kernel

  • Differentiable

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

Removed in version 1.10: Use where() instead, which has the more intuitive argument order: where(cond, value_if_true, value_if_false).

Deprecated since version 1.7.

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

  • Kernel

  • Differentiable

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

Removed in version 1.10: Use where() instead, which has the more intuitive argument order: where(cond, value_if_true, value_if_false).

Deprecated since version 1.7.

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

  • Kernel

  • Differentiable

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

Removed in version 1.10: Use where() instead, which has the more intuitive argument order: where(cond, value_if_true, value_if_false).

Deprecated since version 1.7.

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

  • Kernel

  • Differentiable

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

Removed in version 1.10: Use where() instead, which has the more intuitive argument order: where(cond, value_if_true, value_if_false).

Deprecated since version 1.7.

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

  • Kernel

  • Differentiable

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

Removed in version 1.10: Use where() instead, which has the more intuitive argument order: where(cond, value_if_true, value_if_false).

Deprecated since version 1.7.

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

  • Kernel

  • Differentiable

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

Removed in version 1.10: Use where() instead, which has the more intuitive argument order: where(cond, value_if_true, value_if_false).

Deprecated since version 1.7.

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

  • Kernel

  • Differentiable

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

Removed in version 1.10: Use where() instead, which has the more intuitive argument order: where(cond, value_if_true, value_if_false).

Deprecated since version 1.7.

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

  • Kernel

  • Differentiable

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

Removed in version 1.10: Use where() instead, which has the more intuitive argument order: where(cond, value_if_true, value_if_false).

Deprecated since version 1.7.

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

  • Kernel

  • Differentiable

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

Removed in version 1.10: Use where() instead, which has the more intuitive argument order: where(arr, value_if_true, value_if_false).

Deprecated since version 1.7.