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 mini-batch 张量或大小相同的图像列表。
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 是从张量计算的。
scale_each (bool, 可选) – 如果为
True，则单独缩放图像批次中的每张图像，而不是对所有图像进行 (min, max) 缩放。默认值：False。pad_value (float, 可选) – 填充像素的值。默认值：
0。
- 返回:
包含图像网格的张量。
- 返回类型:
grid (Tensor)
使用
make_grid的示例
