Arranges a batch of (image) tensors in a grid, with optional padding between images. Expects a 4d mini-batch tensor of shape (B x C x H x W).
vision_make_grid( tensor, scale = TRUE, num_rows = 8, padding = 2, pad_value = 0 )
tensor | tensor to arrange in grid. |
---|---|
scale | whether to normalize (min-max-scale) the input tensor. |
num_rows | number of rows making up the grid (default 8). |
padding | amount of padding between batch images (default 2). |
pad_value | pixel value to use for padding. |