| commit | 0ca3ca302ec7908579f1824d8d8a942836996c1c | [log] [tgz] |
|---|---|---|
| author | Trevor Killeen <killeent@users.noreply.github.com> | Thu Aug 17 16:25:33 2017 -0400 |
| committer | Soumith Chintala <soumith@gmail.com> | Tue Sep 05 17:48:55 2017 -0400 |
| tree | deee150af44f70982008b5edd8f6f962481a3b2b | |
| parent | 52e0816bedf11382e988f313acaa0493938e2dd7 [diff] |
test for primspec for concat (#77)
diff --git a/torch/autograd/_functions/tensor.py b/torch/autograd/_functions/tensor.py index 1c214ba..5738c7f 100644 --- a/torch/autograd/_functions/tensor.py +++ b/torch/autograd/_functions/tensor.py
@@ -321,7 +321,7 @@ @staticmethod def primspec(dim, *inputs): - return torch.toffee.op("Concat", inputs, axis=dim) + return torch.toffee.op("Concat", *inputs, axis=dim) @staticmethod def forward(ctx, dim, *inputs):