def set_multiarray_shape()

in src/exporters/coreml/convert.py [0:0]

• 4 lines of code
• 2 McCabe index (conditional complexity)

def set_multiarray_shape(node, shape):
    """Change the shape of the specified input or output in the Core ML model."""
    del node.type.multiArrayType.shape[:]
    for x in shape:
        node.type.multiArrayType.shape.append(x)