tensor_values_are_in_set

sionna.phy.utils.tensor_values_are_in_set(tensor: torch.Tensor, admissible_set: torch.Tensor | List[Any]) → torch.Tensor

Checks if the input tensor values are contained in the specified admissible_set.

Parameters:

• tensor (torch.Tensor) – Tensor to validate

• admissible_set (torch.Tensor | List[Any]) – Set of valid values that the input tensor must be composed of

Outputs:

result – Returns True if and only if tensor values are contained in admissible_set

Examples

import torch
from sionna.phy.utils import tensor_values_are_in_set

tensor = torch.tensor([[1, 0], [0, 1]])

print(tensor_values_are_in_set(tensor, [0, 1, 2]).item())
# True

print(tensor_values_are_in_set(tensor, [0, 2]).item())
# False