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