Commit 69cab8ac authored by rusty1s's avatar rusty1s
Browse files

typo

parent 301ecec5
...@@ -58,7 +58,7 @@ def cat(tensors, dim): ...@@ -58,7 +58,7 @@ def cat(tensors, dim):
if has_colcount: if has_colcount:
colcounts += [tensor.storage.colcount] colcounts += [tensor.storage.colcount]
if has colptr: if has_colptr:
colptr = tensor.storage.colptr colptr = tensor.storage.colptr
colptr = colptr if len(colptrs) == 0 else colptr[1:] colptr = colptr if len(colptrs) == 0 else colptr[1:]
colptrs += [colptr + nnzs] colptrs += [colptr + nnzs]
...@@ -125,16 +125,11 @@ def cat(tensors, dim): ...@@ -125,16 +125,11 @@ def cat(tensors, dim):
old_storage = tensors[0].storage old_storage = tensors[0].storage
storage = old_storage.storage.__class__( storage = old_storage.storage.__class__(
tensors[0].storage.index, tensors[0].storage.index, value=torch.cat(values, dim=dim - 1),
value=torch.cat(values, dim=dim - 1), sparse_size=sparse_size, rowcount=old_storage._rowcount,
sparse_size=sparse_size, rowptr=old_storage._rowcount, colcount=old_storage._rowcount,
rowcount=old_storage._rowcount, colptr=old_storage._rowcount, csr2csc=old_storage._csr2csc,
rowptr=old_storage._rowcount, csc2csr=old_storage._csc2csr, is_sorted=True)
colcount=old_storage._rowcount,
colptr=old_storage._rowcount,
csr2csc=old_storage._csr2csc,
csc2csr=old_storage._csc2csr,
is_sorted=True)
else: else:
raise IndexError( raise IndexError(
......
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