提交 86c195d2 authored 作者: Pascal Lamblin's avatar Pascal Lamblin 提交者: GitHub

Merge pull request #6130 from ChongWu-Biostat/patch-1

Correct a typo in make_node function (tensore.nnet.SoftmaxWithBias)
......@@ -65,7 +65,7 @@ class SoftmaxWithBias(gof.Op):
or x.type.dtype not in tensor.float_dtypes:
raise ValueError('x must be 2-d tensor of floats')
if b.type.ndim != 1 \
or x.type.dtype not in tensor.float_dtypes:
or b.type.dtype not in tensor.float_dtypes:
raise ValueError('b must be 1-d tensor of floats')
sm = x.type()
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论