Permute

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

此模組返回一個張量輸入的檢視，其維度已被重排。

引數：

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

forward(x: Tensor) → Tensor

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

應由所有子類重寫。

注意

儘管前向傳播 (forward pass) 的方法需要在該函式中定義，但應在其後呼叫 Module 例項，而不是直接呼叫此函式，因為前者會負責執行註冊的鉤子 (hooks)，而後者會靜默忽略它們。