def extract(node):
axis = np.array(onnx_attr(node, 'axes', 'ints', default=[]),
dtype=np.int64)
attrs = {'unsqueeze_dims': axis if len(axis) != 0 else None}
# update the attributes of the node
Unsqueeze.update_node_stat(node, attrs)
return __class__.enabled
def extract(cls, node: Node):
Unsqueeze.update_node_stat(node, {})
return cls.enabled
def extract(node: Node):
Unsqueeze.update_node_stat(node, {})
return __class__.enabled