Python Flatten.update_node_stat示例

示例#1

    def extract(cls, node):
        attrs = {
            'axis': 1,
            'end_axis': -1,
        }

        Flatten.update_node_stat(node, attrs)
        return cls.enabled

示例#2

    def extract(node):
        attrs = {
            'axis': 1,
            'end_axis': -1,
        }

        Flatten.update_node_stat(node, attrs)
        return __class__.enabled

示例#3

    def extract(cls, node):
        proto_layer = node.pb
        param = proto_layer.flatten_param

        attrs = {
            'axis': param.axis,
            'end_axis': param.end_axis,
        }

        Flatten.update_node_stat(node, attrs)
        return cls.enabled

示例#4

 def extract(cls, node):
     attrs = {
         'axis': node.module.axis,
     }
     Flatten.update_node_stat(node, attrs)
     return cls.enabled