在IOS中切片并重塑MLMultiArray

时间:2018-04-24 21:14:38

标签: ios coreml coremltools

我有一个coreML模型(从Keras转换而来),它产生的输出为MLMultiArray大小(1,40,30,90)(例如name:finalOutput)。该阵列是通过重塑尺寸的内部张量(1,10800,10)

获得的

我无法将 finalOutput multiArray切成三个维度片段(1,40,30,0:45),(1,40,30, 45:54)和(1,40,30,54:90)。

然后将这些重新塑造成三个维度子阵列 (1,10800,5),(1,10800)amnd(1,10800,4)。

我是iOS世界中快速编程的新手。我可以轻松地在python中做到这一点而没有问题。

在Swift for iOS中,SLICE多阵列和RESHAPE multiArrays是如何实现的?

感谢您的帮助。

2 个答案:

答案 0 :(得分:1)

MLMultiArray上没有简单的方法可用于切片和整形。也许我们在iOS 12中会有这样的东西,但暂时你不得不使用指针算法来做这类事情。

答案 1 :(得分:0)

看看这个example,它使用指针来重塑