Coreml：CustomLayer：Swift：CpuGpu选项将输入清零

问题描述

我为我的coreml模型的自定义层实现了一个快捷类。我使用以下代码指定MLComputeUnits：

     let config = MLModelConfiguration()
    config.computeUnits = MLComputeUnits.cpuAndGPU
    guard let model = try? my_model(configuration: config) else {
      fatalError("Can't init CoreML model")
    }

但是，用于客户层实现的evaluate函数未通过inputs参数获取有效数据。 输入数据全为零。

func evaluate(inputs: [MLMultiArray],outputs: [MLMultiArray]) throws {

如果将MLComputeUnits从cpuAndGPU更改为cpuOnly，则自定义层实现会从inputs函数的evaluate参数接收有效数据。

如果计算单位为cpuOnly

，请告诉我是否需要执行一些其他操作。

解决方法

暂无找到可以解决该程序问题的有效方法，小编努力寻找整理中！

如果你已经找到好的解决方法，欢迎将解决方案带上本链接一起发送给小编。

小编邮箱:dio#foxmail.com (将#修改为@）