test: Test concatenation code

shahrukhqasim · Nov 20, 2017 · 49abb5a · 49abb5a
1 parent 6e310cf
commit 49abb5a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/explore/nnconcat.py b/python/explore/nnconcat.py
@@ -7,7 +7,7 @@ def expand_way_2(A):
     return A.expand(10,10,2).contiguous().view(100,2)
 
 
-A = torch.cat((torch.unsqueeze(torch.range(1,10), dim=1), torch.unsqueeze(torch.range(1,10), dim=1)), dim=1)
+A = torch.cat((torch.unsqueeze(torch.range(1,10), dim=1), torch.unsqueeze(2*torch.range(1,10), dim=1)), dim=1)
 print(A)
 A1=expand_way_1(A)
 A2=expand_way_2(A)