reference/generated/tensordict.is_integer

Run in Google Colab

Colab

Download Notebook

Notebook

View on GitHub

GitHub

is_integer

tensordict.is_integer(*args) → bool

Whether the last argument is an integer tensor (signed or unsigned, not bool).

The predicate tests its last argument only, so it can be passed as fn(value) and as fn(key, value): see is_inexact().

Parameters:

*args – the value to test is the last one. A key may come before it.

Returns:

True if the dtype of the value is neither floating-point, complex nor torch.bool.

Examples

>>> import torch
>>> from tensordict import is_integer
>>> is_integer(torch.zeros(3, dtype=torch.uint8)), is_integer(torch.zeros(3, dtype=torch.bool))
(True, False)