Permute

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

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

引數:

dims (List[int]) – 期望的維度順序

forward(x: Tensor) → Tensor

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

所有子類都應重寫此方法。

注意

儘管前向傳播的實現需要在此函式中定義，但您應該在之後呼叫 Module 例項而不是此函式，因為前者會處理註冊的鉤子，而後者則會靜默忽略它們。