提交 48e71079 authored 作者: Amjad Almahairi's avatar Amjad Almahairi

return ints

上级 60b3686f
......@@ -245,7 +245,7 @@ class WeightedSelectionFromUniform(Op):
unis.shape[0], pvals.shape[0], n_samples)
if z[0] is None or not numpy.all(z[0].shape == [pvals.shape[0], n_samples]):
z[0] = -1 * numpy.ones((pvals.shape[0], n_samples), dtype=node.outputs[0].dtype)
z[0] = -1 * numpy.ones((pvals.shape[0], n_samples), dtype='int')
nb_multi = pvals.shape[0]
nb_outcomes = pvals.shape[1]
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论