def extract(cls, node):
axis = np.array(onnx_attr(node, 'axes', 'ints', default=[]),
dtype=np.int64)
ExpandDims.update_node_stat(node, {'expand_axis': axis})
return cls.enabled
def extract(cls, node):
attrs = get_mxnet_layer_attrs(node.symbol_dict)
expand_axis = attrs.int('axis', None)
ExpandDims.update_node_stat(node, {'expand_axis': expand_axis})
return cls.enabled