• Pedro Cuenca's avatar
    Use ONNX / Core ML compatible method to broadcast (#310) · e49dd03d
    Pedro Cuenca authored
    * Use ONNX / Core ML compatible method to broadcast.
    
    Unfortunately `tile` could not be used either, it's still not compatible
    with ONNX.
    
    See #284.
    
    * Add comment about why broadcast_to is not used.
    
    Also, apply style to changed files.
    
    * Make sure broadcast remains in same device.
    e49dd03d
unet_2d.py 7.03 KB