提交 ac631700 authored 作者: Frederic's avatar Frederic

Add name to variable

上级 20084f74
......@@ -3010,7 +3010,7 @@ def mean(input, axis=None, dtype=None, op=False, keepdims=False,
if dtype == 'float16' or (dtype is None and input.dtype == 'float16'):
s = cast(s, 'float16')
s.name = 'mean'
return s
......@@ -3055,7 +3055,9 @@ def var(input, axis=None, keepdims=False):
centered_input = input - mean_input
# return the mean sqr
return mean((centered_input ** 2), axis, keepdims=keepdims)
v = mean((centered_input ** 2), axis, keepdims=keepdims)
v.name = 'var'
return v
@constructor
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论