OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-07 12:21:27 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity

fix legacy comment after variable tensor merge (#7771)

Browse Source
This commit is contained in:
Tongzhou Wang 2018-05-22 19:08:42 -04:00 committed by Soumith Chintala
parent 60745b3380
commit ee628d64b9
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
torch/nn/parallel/scatter_gather.py
View File

@ -6,8 +6,7 @@ def scatter(inputs, target_gpus, dim=0):
r"""
Slices tensors into approximately equal chunks and
distributes them across given GPUs. Duplicates
references to objects that are not tensors. Does not
support Tensors.
references to objects that are not tensors.
"""
def scatter_map(obj):
if isinstance(obj, torch.Tensor):