Commit 481e81fe authored by rusty1s's avatar rusty1s
Browse files

fix view on zero-element tensors

parent b98d1f3b
from itertools import repeat
import torch
def maybe_dim_size(index, dim_size=None):
if dim_size is not None:
......@@ -14,7 +16,10 @@ def gen(src, index, dim=-1, out=None, dim_size=None, fill_value=0):
if index.dim() == 1:
index_size = list(repeat(1, src.dim()))
index_size[dim] = src.size(dim)
if index.numel() > 0:
index = index.view(index_size).expand_as(src)
else: # PyTorch has a bug when view is used on zero-element tensors.
index = src.new_empty(index_size, dtype=torch.long)
# Generate output tensor if not given.
if out is None:
......
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