get_dimensions

torchvision.transforms.functional.get_dimensions(img: Tensor) → list[int]

返回影像的尺寸，格式為 [通道數, 高度, 寬度]。

引數:

img (PIL Image 或 Tensor) – 要檢查的影像。

返回:

影像的尺寸。

返回型別:

List[int]