ios - 如何在 MLMultiarray 元素上应用 Sigmoid？

Question

对于Python中的 torch 张量，我可以使用
y = x[i].sigmoid()

如何在 Swift 中使用 MLMultiArray 复制此内容？
我可以通过 VNCoreMLFeatureValueObservation.featureValue.multiArrayValue 获取数组
现在我想对此应用 sigmoid

我使用的 sigmoid 函数是:

func sigmoid(z: Double) -> Double {
        return 1.0 / (1.0 + exp(-z))
    }

Answer 1

从内存中输入此内容，因此可能会出现错误，但基本上您会获取指向内存的指针，然后循环遍历它:

let ptr = multiArray.dataPointer.assumingMemoryBound(to: Double.self)
for i in 0..<multiArray.count {
  ptr[i] = sigmoid(ptr[i])
}

但是，将 sigmoid 操作添加到 Core ML 模型本身并让 Core ML 操心要简单得多。您可以通过在转换之前或之后向原始模型添加 sigmoid 操作来完成此操作(有关如何执行此操作的说明，请参阅我的书《Core ML Survival Guide》)。