如何在 CoreML 中访问 MLMultiArray 中的元素

Question

我已经使用 initWithDataPointer 初始化了 MLMultiArray，如下面的代码所示：

float count = 512 * 384;
  double *tempBuffer = malloc(count * sizeof(double));
  NSError *error = NULL;
  NSArray *shape = [NSArray arrayWithObjects:[NSNumber numberWithInt:1],[NSNumber numberWithInt:512],[NSNumber numberWithInt:384], nil];
  NSArray *stride = [NSArray arrayWithObjects:[NSNumber numberWithInt:1],[NSNumber numberWithInt:1],[NSNumber numberWithInt:1], nil];

  MLMultiArray *mlMultiArray = [[MLMultiArray alloc] initWithDataPointer:tempBuffer
                                                                   shape:shape
                                                                dataType:MLMultiArrayDataTypeDouble
                                                                 strides:stride
                                                             deallocator:^(void * _Nonnull bytes) { free(bytes); }
                                                                   error:&error];

根据此 link 中提到的 MLMultiArray 文档，需要使用 subscript 来访问元素。

如果我按照所示的方式访问元素，是否正确？

NSNumber *val = [mlMultiArray objectForKeyedSubscript:[NSArray arrayWithObjects:[NSNumber numberWithInt:1],[NSNumber numberWithInt:1],[NSNumber numberWithInt:1], nil]];

Answer 1

我建议你使用mlMultiArray.dataPointer，转换为double *，然后直接访问数据缓冲区的内容。您可以使用步幅计算元素 i, j, k 的位置：

double *ptr = (double *) mlMultiArray.dataPointer;
NSInteger offset = i*stride[0].intValue + j*stride[1].intValue + k*stride[2].intValue;
double val = ptr[offset];

如何在 CoreML 中访问 MLMultiArray 中的元素

How to access elements inside MLMultiArray in CoreML

machine-learning

objective-c

multidimensional-array

ios

coreml