提交 83e3cc9e authored 作者: James Bergstra's avatar James Bergstra

modified shared constructors broadcasting default to match TensorType

上级 d5d58e3c
...@@ -79,7 +79,7 @@ def shared_constructor(value, name, strict=False, broadcastable=None): ...@@ -79,7 +79,7 @@ def shared_constructor(value, name, strict=False, broadcastable=None):
raise TypeError('float32 ndarray required') raise TypeError('float32 ndarray required')
if broadcastable is None: if broadcastable is None:
broadcastable = [b==1 for b in value.shape] broadcastable = (False,) * len(value.shape)
type = CudaNdarrayType(broadcastable=broadcastable) type = CudaNdarrayType(broadcastable=broadcastable)
return CudaNdarraySharedVariable(type=type, value=_value, name=name, strict=strict) return CudaNdarraySharedVariable(type=type, value=_value, name=name, strict=strict)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论