UnsafeBufferPointer's extension that takes MLMultiArray has been deprecated. You can get a contiguous array of a shaped array as follows.
if let shapedArrayValue = prediction.featureValue(for: "output")?.shapedArrayValue(of: Float.self) {
result = shapedArrayValue.scalars
}
As for the performance, could you use Instruments to find out the bottlenecks? You may find this video useful.
https://developer.apple.com/videos/play/wwdc2022/10027/
moto_coreml_apple
Users receive reputation points by contributing quality content to the forums.
Posts include post and replies published on the forums.
Post authors can mark replies as Accepted to indicate successful solutions.
Apple Developer Forums admins can mark replies as Apple Recommended to indicate an approved solution