调整核心模型的输入

Question

我有一个 coremlmodel 接受形状MultiArray (Float32 67 x 256 x 320)的输入 我很难塑造这个模型的输入。 目前，我正在努力实现它，

var m = try! MLMultiArray(shape: [67,256,320], dataType: .double)
for i in 0...66{           
var cost = rand((256,320)) // this is coming from swix [SWIX]
memcpy(m.dataPointer+i*256*320, &cur_cost.flat.grid , 256*320)
}

稍后我将不得不用该大小的矩阵替换兰特。我首先将其用于测试目的。

任何关于如何塑造输入以适应体积的指示将不胜感激。

[SWIX]

Answer 1

代码中的错误似乎是您正在复制字节而不是双精度。双精度为 8 个字节，因此您的偏移量应i*256*320*MemoryLayout<Double>.stride，并且要复制的数量应256*320*MemoryLayout<Double>.stride。

请注意，您还可以使用 MLMultiArray 的strides属性来计算数组中给定数据元素的偏移量：

let offset = i0 * strides[0].intValue + i1 * strides[1].intValue + i2 * strides[2].intValue