Commit c94ea5ff authored by Alykhan Tejani's avatar Alykhan Tejani Committed by Francisco Massa
Browse files

update docs for nrows param in make_grid (#250)

parent dd53c974
......@@ -10,8 +10,8 @@ def make_grid(tensor, nrow=8, padding=2,
Args:
tensor (Tensor or list): 4D mini-batch Tensor of shape (B x C x H x W)
or a list of images all of the same size.
nrow (int, optional): Number of rows in grid. Final grid size is
(B / nrow, nrow). Default is 8.
nrow (int, optional): Number of images displayed in each row of the grid.
The Final grid size is (B / nrow, nrow). Default is 8.
padding (int, optional): amount of padding. Default is 2.
normalize (bool, optional): If True, shift the image to the range (0, 1),
by subtracting the minimum and dividing by the maximum pixel value.
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment