提交 9d21d6c0 authored 作者: Chong Wu's avatar Chong Wu 提交者: GitHub

Correct a typo in SoftmaxWithBias (make_node)

I found a typo in make_node function. We should correct it otherwise some potential wired bugs may occur.
上级 e2ee93fc
...@@ -65,7 +65,7 @@ class SoftmaxWithBias(gof.Op): ...@@ -65,7 +65,7 @@ class SoftmaxWithBias(gof.Op):
or x.type.dtype not in tensor.float_dtypes: or x.type.dtype not in tensor.float_dtypes:
raise ValueError('x must be 2-d tensor of floats') raise ValueError('x must be 2-d tensor of floats')
if b.type.ndim != 1 \ 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') raise ValueError('b must be 1-d tensor of floats')
sm = x.type() sm = x.type()
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论