如何在 Swift 中将 NSArray 复制到 MLMultiArray 中？

发布于 2025-01-16 10:11:11 字数 666 浏览 6 评论 0原文

我有一个 plist 文件中大小为 (1, 5, 401) 的 NSArray。我想将其复制到相同形状的 MLMultiArray 中，然后将其传递到 Swift 中的 CoreML 模型。如何将 NSArray 中的值复制到 MLMultiArray？

let filepath = Bundle.main.path(forResource:"testPlist", ofType: "plist")
let dict2 = NSDictionary(contentsOfFile: filepath!)

let test_input:NSArray = dict2?.object(forKey:"test_input") as! NSArray

let mlMultiArray_input = try? MLMultiArray (shape:[1,5,401],dataType : MLMultiArrayDataType.double)

plist 文件：

原文

I have an NSArray of size (1, 5, 401) from a plist file. I want to copy that into MLMultiArray of the same shape and then pass it to a CoreML model in Swift. How can I copy the values from NSArray to MLMultiArray?

let filepath = Bundle.main.path(forResource:"testPlist", ofType: "plist")
let dict2 = NSDictionary(contentsOfFile: filepath!)

let test_input:NSArray = dict2?.object(forKey:"test_input") as! NSArray

let mlMultiArray_input = try? MLMultiArray (shape:[1,5,401],dataType : MLMultiArrayDataType.double)

plist file:

分享到QQ

分享到微博