make_grid¶
- torchvision.utils.make_grid(tensor: Union[Tensor, List[Tensor]], nrow: int = 8, padding: int = 2, normalize: bool = False, value_range: Optional[Tuple[int, int]] = None, scale_each: bool = False, pad_value: float = 0.0) Tensor [source]¶
制作一个图片网格。
- Parameters:
tensor (Tensor 或 list) – 形状为 (B x C x H x W) 的4D小批量张量 或大小相同的图像列表。
nrow (int, 可选) – 网格中每行显示的图像数量。 最终的网格大小为
(B / nrow, nrow)
。默认值:8
。padding (int, 可选) – 填充量。默认值:
2
。normalize (bool, 可选) – 如果为True,则将图像移动到范围(0, 1), 通过
value_range
指定的最小值和最大值。默认值:False
。value_range (tuple, optional) – 元组 (min, max),其中 min 和 max 是数字, 然后这些数字用于归一化图像。默认情况下,min 和 max 是从张量中计算得出的。
scale_each (bool, 可选) – 如果为
True
,则分别缩放批次中的每个图像,而不是所有图像的(最小,最大)值。默认值:False
。pad_value (float, 可选) – 填充像素的值。默认值:
0
。
- Returns:
包含图像网格的张量。
- Return type:
网格 (张量)
使用
make_grid
的示例: