如何在iOS中使用Tensorflow训练的机器学习模型

import TensorFlowLite

class AppDelegate: UIResponder, UIApplicationDelegate {
    func application(_ application: UIApplication, didFinishLaunchingWithOptions launchOptions: [UIApplication.LaunchOptionsKey: Any]?) -> Bool {
        // 初始化TensorFlow库
        TensorFlowLite.initialize()
        return true
    }
}

import TensorFlowLite

// 加载模型
guard let modelPath = Bundle.main.path(forResource: "model", ofType: "tflite") else {
    fatalError("模型文件未找到")
}
guard let interpreter = try? Interpreter(modelPath: modelPath) else {
    fatalError("模型加载失败")
}

// 准备输入数据
let input: [Float32] = [1.0, 2.0, 3.0]
let inputTensor = try? interpreter.input(at: 0)
try? inputTensor?.copyData(from: input)

// 运行模型
try? interpreter.invoke()

// 获取输出结果
let outputTensor = try? interpreter.output(at: 0)
let output = outputTensor?.data.toArray(type: Float32.self)
print(output)