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