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