any

nvtripy.any(input: Tensor, dim: int | Sequence[int] | None = None, keepdim: bool = False) → Tensor[source]

Returns a new tensor containing the logical OR of the elements of the input tensor along the specified dimension.

Parameters:

  • input (Tensor) – The input tensor.

  • dim (int | Sequence[int] | None) – The dimension or dimensions along which to reduce. If this is not provided, all dimensions are reduced.

  • keepdim (bool) – Whether to retain reduced dimensions in the output. If this is False, reduced dimensions will be squeezed.

Returns:

A new bool tensor.

Return type:

Tensor

INPUT REQUIREMENTS:

input.dtype == bool

OUTPUT GUARANTEES:

return[0].dtype == bool

Example
1input = tp.Tensor([True, False])
2out = tp.any(input)
Local Variables
>>> input
tensor([True, False], dtype=bool, loc=cpu:0, shape=(2,))

>>> out
tensor(1, dtype=bool, loc=gpu:0, shape=())