capture_non_tensor_stack

class tensordict.capture_non_tensor_stack(allow_none=False)

獲取當前捕獲非張量堆疊的設定。

引數:

allow_none (bool, optional) – 如果為 True，則在未指定任何設定時返回 None。否則，返回預設設定。預設為 False。

另請參閱： set_capture_non_tensor_stack()

返回:

捕獲非張量堆疊的當前設定。

返回型別:

bool 或 None