Permute

class torchvision.ops.Permute(dims: List[int])

此模組傳回張量輸入的視圖，其維度已排列。

參數:

dims (List[int]) – 維度的所需順序

forward(x: Tensor) → Tensor

定義每次呼叫時執行的計算。

應由所有子類別覆寫。

注意

雖然前向傳遞的步驟需要在這個函式中定義，但之後應該呼叫 Module 實例而不是這個函式，因為前者會處理已註冊的鉤子，而後者會靜默地忽略它們。