make_grid¶
- torchvision.utils.make_grid(tensor: Union[Tensor, list[torch.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[源代码]¶
创建图像网格。
- 参数:
tensor (Tensor 或 list) – 形状为 (B x C x H x W) 的 4D 小批量 Tensor 或一个包含大小相同的图像的列表。
nrow (int, 可选) – 网格中每行显示的图像数量。最终网格大小为
(B / nrow, nrow)。默认为8。padding (int, 可选) – 填充量。默认为
2。normalize (bool, 可选) – 如果为 True，则根据
value_range指定的最小值和最大值，将图像移至 (0, 1) 的范围。默认为False。value_range (tuple, 可选) – 元组 (min, max)，其中 min 和 max 是数字，则这些数字用于归一化图像。默认情况下，min 和 max 从 tensor 计算得出。
scale_each (bool, 可选) – 如果为
True，则分别对批量中的每个图像进行缩放，而不是对所有图像的 (min, max) 进行缩放。默认为False。pad_value (float, 可选) – 填充像素的值。默认为
0。
- 返回:
包含图像网格的 tensor。
- 返回类型:
grid (Tensor)
使用
make_grid的示例
