is_tensor_collection

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

檢查資料物件或類型是否為來自 tensordict 函式庫的 tensor 容器。

傳回:

如果輸入是 TensorDictBase 子類別、tensorclass 或這些的實例，則True。否則為 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