diff options
author | soumith <soumith@fb.com> | 2016-08-12 05:13:31 +0300 |
---|---|---|
committer | soumith <soumith@fb.com> | 2016-08-12 05:13:31 +0300 |
commit | 9d5b1b1026a559b0c4f6e36432db53c3dd002c24 (patch) | |
tree | 00a203fd3f055ef7f416be50b2fda780d1dee671 /torch | |
parent | 3cd2b3f0d9f5a60491194aeb875c1ebf9525c740 (diff) |
fix a small bug
Diffstat (limited to 'torch')
-rw-r--r-- | torch/generic/Tensor.c | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/torch/generic/Tensor.c b/torch/generic/Tensor.c index fd0251c..4fed963 100644 --- a/torch/generic/Tensor.c +++ b/torch/generic/Tensor.c @@ -423,7 +423,7 @@ static int torch_Tensor_(indexSelect)(lua_State *L) dim = luaL_checkint(L, 3) - 1; index = luaT_toudata(L, 4, "torch.CudaLongTensor"); longIndex = luaT_toudata(L, 4, "torch.LongTensor"); - realIndex = luaT_toudata(L, 3, torch_Tensor); + realIndex = luaT_toudata(L, 4, torch_Tensor); if (!index && !longIndex && !realIndex) luaT_typerror(L, 3, "CudaLongTensor | LongTensor | Tensor"); tensor = luaT_checkudata(L,1,torch_Tensor); } |