reference/generated/tensordict.is_non_tensor

Run in Google Colab

Colab

Download Notebook

Notebook

View on GitHub

GitHub

is_non_tensor

tensordict.is_non_tensor(data) → bool

Returns True if data is non-tensor data or a stack of non-tensor data.

Parameters:

data (object) – the object to check, for example a value read from a tensordict with get().

Examples

>>> import torch
>>> from tensordict import NonTensorData, TensorDict, is_non_tensor
>>> td = TensorDict(x=torch.zeros(2), name=NonTensorData("a"))
>>> is_non_tensor(td.get("name")), is_non_tensor(td.get("x"))
(True, False)
>>> td["name"]
'a'