提交 9b71642c authored 作者: Sina Honari's avatar Sina Honari

method name correction

上级 db7ae002
...@@ -4656,7 +4656,7 @@ class Flatten(Op): ...@@ -4656,7 +4656,7 @@ class Flatten(Op):
} }
""" % locals() """ % locals()
def is_flatten(node, outdim=1): def is_flat(node, outdim=1):
""" """
Checks whether node's op is an instance of Reshape Checks whether node's op is an instance of Reshape
and verifies the dimensionality of variable is correct. and verifies the dimensionality of variable is correct.
...@@ -4665,7 +4665,7 @@ def is_flatten(node, outdim=1): ...@@ -4665,7 +4665,7 @@ def is_flatten(node, outdim=1):
node.inputs[1].ndim == outdim node.inputs[1].ndim == outdim
def flat(x, outdim=1): def flatten(x, outdim=1):
""" """
Reshapes the variable x by keeping Reshapes the variable x by keeping
the first outdim-1 dimension(s) of x the same the first outdim-1 dimension(s) of x the same
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论