初次接触CoreML,正在尝试理解一些基本概念。
我正在处理一个输出为以下格式的模型:
float32 [1,896,16]
在使用模型并将输出获取为MLMultiArray
时,我得到以下结果:
let output = prediction.regressors // MLMultiArrayprint(output.debugDescription) // Float32 1 x 896 x 16 arrayprint(output.count) // 14336, 这是896x16
我可以简单地使用output[0]..output[1]..
来访问每个元素
对于存储在MLMultiArray
中的任何数据类型,这都是正确的吗?这是Swift提供给我们的“便利”吗?
如果是扁平数组的情况,它的顺序会与矩阵的顺序相同吗?
回答:
MLMultiArray
是否是模型输出矩阵的扁平数组?不是,它是给定维度的多维数组
您可以像下面这样将其转换为数组,并且它应该与矩阵的顺序相同。
let length = output.count let doublePtr = output.dataPointer.bindMemory(to: Double.self, capacity: length) let doubleBuffer = UnsafeBufferPointer(start: doublePtr, count: length) let outputArray = Array(doubleBuffer)
关于数据类型,苹果的文档仅显示三种类型:这里
case int32
代表多维数组的整数类型,通常用于文本编码。
case float32
代表多维数组中的浮点类型。
case double
代表多维数组的双精度类型。