get_dimensions¶ torchvision.transforms.functional.get_dimensions(img: Tensor) → List[int][源代码]¶ 将图像的尺寸以 [通道数,高度,宽度] 的形式返回。 参数:: img (PIL 图像 或 Tensor) – 要检查的图像。 返回值:: 图像尺寸。 返回类型:: List[int]