is_tensor_collection

class tensordict.is_tensor_collection(datatype: Union[type, Any])

检查一个数据对象或类型是否是 tensordict 库中的张量容器。

返回:

True 如果输入是 TensorDictBase 的子类，tensorclass 或它们的实例。否则为 False。

示例

>>> is_tensor_collection(TensorDictBase)  # True
>>> is_tensor_collection(TensorDict())  # True
>>> @tensorclass
... class MyClass:
...     pass
...
>>> is_tensor_collection(MyClass)  # True
>>> is_tensor_collection(MyClass(batch_size=[]))  # True