Shortcuts

获取维度

torchvision.transforms.functional.get_dimensions(img: Tensor) List[int][source]

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

Parameters:

img (PIL ImageTensor) – 要检查的图像。

Returns:

图像的尺寸。

Return type:

列表[int]