capture_non_tensor_stack

class tensordict.capture_non_tensor_stack(allow_none=False)

Get the current setting for capturing non-tensor stacks.

Parameters:

allow_none (bool, optional) – If True, returns None if no setting has been specified. Otherwise, returns the default setting. Defaults to False.

seealso: set_capture_non_tensor_stack()

Returns:

The current setting for capturing non-tensor stacks.

Return type:

bool or None