@@ -10,16 +10,17 @@ def make_grid(tensor, nrow=8, padding=2,
1010 Args:
1111 tensor (Tensor or list): 4D mini-batch Tensor of shape (B x C x H x W)
1212 or a list of images all of the same size.
13- nrows (int, optional): Number of rows in grid. Final grid size is
13+ nrow (int, optional): Number of rows in grid. Final grid size is
1414 (B / nrow, nrow). Default is 8.
15+ padding (int, optional): amount of padding. Default is 2.
1516 normalize (bool, optional): If True, shift the image to the range (0, 1),
1617 by subtracting the minimum and dividing by the maximum pixel value.
1718 range (tuple, optional): tuple (min, max) where min and max are numbers,
1819 then these numbers are used to normalize the image. By default, min and max
1920 are computed from the tensor.
20- scale_each(bool, optional): If True, scale each image in the batch of
21+ scale_each (bool, optional): If True, scale each image in the batch of
2122 images separately rather than the (min, max) over all images.
22- pad_value(float, optional): Value for the padded pixels.
23+ pad_value (float, optional): Value for the padded pixels.
2324
2425 Example:
2526 See this notebook `here <https://gist.github.com/anonymous/bf16430f7750c023141c562f3e9f2a91>`_
0 commit comments