提交 6ace929c 编写于 作者: W wanghaoshuang

Rename variable name.

上级 52be2a2a
...@@ -3863,17 +3863,17 @@ class SwitchOrderLayer(LayerBase): ...@@ -3863,17 +3863,17 @@ class SwitchOrderLayer(LayerBase):
if reshape is None: if reshape is None:
self.set_layer_size(input_layer.size) self.set_layer_size(input_layer.size)
else: else:
inH = input_layer.height in_h = input_layer.height
inW = input_layer.width in_w = input_layer.width
if input_layer.has_depth(): if input_layer.has_depth():
inD = input_layer.depth in_d = input_layer.depth
inC = input_layer.size / inH / inW / inD in_c = input_layer.size / in_h / in_w / in_d
out_dims = [0, inD, inH, inW, inC] out_dims = [0, in_d, in_h, in_w, in_c]
size = reduce(lambda x, y: x * y, size = reduce(lambda x, y: x * y,
out_dims[reshape['width'][0]:]) out_dims[reshape['width'][0]:])
else: else:
inC = input_layer.size / inH / inW in_c = input_layer.size / in_h / in_w
out_dims = [0, inH, inW, inC] out_dims = [0, in_h, in_w, in_c]
size = reduce(lambda x, y: x * y, size = reduce(lambda x, y: x * y,
out_dims[reshape['width'][0]:]) out_dims[reshape['width'][0]:])
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册