get_dimensions

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

返回图像的尺寸，格式为 [通道数, 高度, 宽度]。

参数:

img (PIL Image 或 Tensor) – 要检查的图像。

返回:

图像尺寸。

返回类型:

List[int]