diff options
author | lichen <lichenarthurdata@gmail.com> | 2017-09-04 14:55:56 +0300 |
---|---|---|
committer | Soumith Chintala <soumith@gmail.com> | 2017-09-04 18:12:52 +0300 |
commit | 75a86469aa9e2f5f04e11895b269ec22eb0e4687 (patch) | |
tree | 7e3de50fdcb0f339bafc2f0215cf985ea4d39510 | |
parent | f5daa71f0541e57e505f1304e3667c3c7be38ccb (diff) |
fix bug for THTensor data access
-rw-r--r-- | lib/TH/generic/THTensorRandom.c | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/lib/TH/generic/THTensorRandom.c b/lib/TH/generic/THTensorRandom.c index 5bfabbc..21359a1 100644 --- a/lib/TH/generic/THTensorRandom.c +++ b/lib/TH/generic/THTensorRandom.c @@ -147,7 +147,7 @@ void THTensor_(multinomialAliasSetup)(THTensor *probs, THLongTensor *J, THTensor THTensor_fastSet1d(J, small, large); q_data[large * q->stride[0]] -= 1.0 - THTensor_fastGet1d(q, small); - if(q_data[large] < 1.0) + if(q_data[large * q->stride[0]] < 1.0) { THTensor_fastSet1d(smaller, small_c-1, large); large_c -= 1; |