Skip to content

GitLab

You need to sign in or sign up before continuing.
Commit de5dcb93 authored by Adam Paszke's avatar Adam Paszke
Browse files

Remove long from make_grid

parent 97871cd9
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
torchvision/utils.py
View file @ de5dcb93
...@@ -25,7 +25,7 @@ def make_grid(tensor, nrow=8, padding=2, ...@@ -25,7 +25,7 @@ def make_grid(tensor, nrow=8, padding=2,
if isinstance(tensor, list): if isinstance(tensor, list):
tensorlist = tensor tensorlist = tensor
numImages = len(tensorlist) numImages = len(tensorlist)
size = torch.Size(torch.Size([long(numImages)]) + tensorlist[0].size()) size = torch.Size(torch.Size([numImages]) + tensorlist[0].size())
tensor = tensorlist[0].new(size) tensor = tensorlist[0].new(size)
for i in irange(numImages): for i in irange(numImages):
tensor[i].copy_(tensorlist[i]) tensor[i].copy_(tensorlist[i])
......
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