ios - 在 IOS 中切片和 reshape MLMultiArray

Question

我有一个 coreML 模型(从 Keras 转换而来)，它以大小为 (1,40,30,90) 的 MLMultiArray 形式生成输出(例如名称:finalOutput)。该数组是通过 reshape 维度为 (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) 和 (1,10800,4)。

我是 iOS 世界中快速编程的新手。我可以轻松地在 python 中完成此操作，没有任何问题。

iOS 版 Swift 中的 SLICE 多数组和 RESHAPE 多数组是如何实现的？

感谢您的帮助。

Answer 1

MLMultiArray 上没有简单的方法来进行切片和整形。也许我们会在 iOS 12 中进行类似的操作，但目前您必须使用指针算术来完成此类操作。