快捷键

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 的示例

可视化实用程序

可视化实用程序

文档

访问 PyTorch 的综合开发者文档

查看文档

教程

获取针对初学者和高级开发人员的深入教程

查看教程

资源

查找开发资源并获得问题的解答

查看资源