cmotl
diff --git a/‎lib/TH/generic/THTensorRandom.c‎
Lines changed: 1 addition & 1 deletion b/‎lib/TH/generic/THTensorRandom.c‎
Lines changed: 1 addition & 1 deletion
@@ -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;
Original file line number	Diff line number	Diff line change
`@@ -147,7 +147,7 @@ void THTensor_(multinomialAliasSetup)(THTensor probs, THLongTensor J, THTensor`
`147`	`147`	`THTensor_fastSet1d(J, small, large);`
`148`	`148`	`q_data[large * q->stride[0]] -= 1.0 - THTensor_fastGet1d(q, small);`
`149`	`149`
`150`		`- if(q_data[large] < 1.0)`
	`150`	`+ if(q_data[large * q->stride[0]] < 1.0)`
`151`	`151`	`{`
`152`	`152`	`THTensor_fastSet1d(smaller, small_c-1, large);`
`153`	`153`	`large_c -= 1;`