快捷方式

list_to_stack

class tensordict.list_to_stack(allow_none=False)

檢索 TensorDict 中列表到堆疊轉換的當前設定。

此函式檢查全域性環境變數或上下文管理器設定,以確定在分配給 TensorDict 時是否應自動堆疊列表。

當前行為
返回列表到堆疊轉換的當前設定。如果未定義該設定且 allow_none

為 True,則返回 None。否則,返回預設設定。

引數:

allow_none (bool) – 如果為 True,則允許在未定義設定時返回 None。

返回:

列表到堆疊轉換的當前設定。

返回型別:

bool 或 None

另請參閱

set_list_to_stack.

文件

訪問全面的 PyTorch 開發者文件

檢視文件

教程

為初學者和高階開發者提供深入的教程

檢視教程

資源

查詢開發資源並讓您的問題得到解答

檢視資源