Skip to content

GitLab

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
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
torchvision/utils.py
View file @ c94ea5ff
...@@ -10,8 +10,8 @@ def make_grid(tensor, nrow=8, padding=2, ...@@ -10,8 +10,8 @@ def make_grid(tensor, nrow=8, padding=2,
Args: Args:
tensor (Tensor or list): 4D mini-batch Tensor of shape (B x C x H x W) 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. or a list of images all of the same size.
nrow (int, optional): Number of rows in grid. Final grid size is nrow (int, optional): Number of images displayed in each row of the grid.
(B / nrow, nrow). Default is 8. The Final grid size is (B / nrow, nrow). Default is 8.
padding (int, optional): amount of padding. Default is 2. padding (int, optional): amount of padding. Default is 2.
normalize (bool, optional): If True, shift the image to the range (0, 1), normalize (bool, optional): If True, shift the image to the range (0, 1),
by subtracting the minimum and dividing by the maximum pixel value. 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