Python PyTorch Tensor.align_to用法及代碼示例

本文簡要介紹python語言中 torch.Tensor.align_to 的用法。

用法:

align_to(*names)

參數：

names(str的可迭代) -輸出張量的所需維度排序。最多可以包含一個省略號，該省略號擴展到 self 的所有未提及的 dim 名稱。

排列 self 張量的維度以匹配 names 中指定的順序，並為任何新名稱添加 size-one 變暗。

self 的所有暗淡必須命名才能使用此方法。結果張量是原始張量的視圖。

self 的所有維度名稱必須出現在 names 中。 names 可能包含 self.names 中沒有的其他名稱；對於每個新名稱，輸出張量都有一個 size-one 維度。

names 最多可包含一個省略號 (... )。省略號將擴展為等於 self 中未在 names 中提及的所有維度名稱，按照它們在 self 中出現的順序排列。

Python 2 不支持省略號，但可以使用字符串文字代替 ('...')。

例子：

>>> tensor = torch.randn(2, 2, 2, 2, 2, 2)
>>> named_tensor = tensor.refine_names('A', 'B', 'C', 'D', 'E', 'F')

# Move the F and E dims to the front while keeping the rest in order
>>> named_tensor.align_to('F', 'E', ...)

警告

命名張量 API 是實驗性的，可能會發生變化。