get_dimensions¶ torchvision.transforms.functional.get_dimensions(img: Tensor) → List[int][source]¶ 返回图像的维度，格式为 [通道数, 高, 宽]。 参数: img (PIL Image or Tensor) – 要检查的图像。 返回值: 图像维度。 返回类型: List[int]
