提交 9e439d5f authored 作者: Frederic's avatar Frederic

Revert variable name change as this break user interface and numpy compatibility

上级 c331362f
...@@ -2016,26 +2016,26 @@ def zeros_like(model, dtype=None): ...@@ -2016,26 +2016,26 @@ def zeros_like(model, dtype=None):
return fill(model, constant(0.0, dtype=dtype)) return fill(model, constant(0.0, dtype=dtype))
def zeros(shp, dtype=None): def zeros(shape, dtype=None):
""" """
Create a Tensor filled with zeros, closer to Numpy's syntax than ``alloc``. Create a Tensor filled with zeros, closer to Numpy's syntax than ``alloc``.
""" """
if not isinstance(shp, (list, tuple, TensorVariable)): if not isinstance(shape, (list, tuple, TensorVariable)):
shp = [shp] shape = [shape]
if dtype is None: if dtype is None:
dtype = config.floatX dtype = config.floatX
return alloc(numpy.array(0, dtype=dtype), *shp) return alloc(numpy.array(0, dtype=dtype), *shape)
def ones(shp, dtype=None): def ones(shape, dtype=None):
""" """
Create a Tensor filled with ones, closer to Numpy's syntax than ``alloc``. Create a Tensor filled with ones, closer to Numpy's syntax than ``alloc``.
""" """
if not isinstance(shp, (list, tuple, TensorVariable)): if not isinstance(shape, (list, tuple, TensorVariable)):
shp = [shp] shape = [shape]
if dtype is None: if dtype is None:
dtype = config.floatX dtype = config.floatX
return alloc(numpy.array(1, dtype=dtype), *shp) return alloc(numpy.array(1, dtype=dtype), *shape)
class Nonzero(gof.Op): class Nonzero(gof.Op):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论