Core MLのモデルの出力がMLMultiArrayの場合のメモ

画像分類みたいに出力が単純なクラスのインデックスではなく、画像を出力するようなCore MLモデルではよく出力の型がMLMultiArrayであることが多い。

たとえば公式のデプス推定モデル"FCRN-DepthPrediction"のOutputsのTypeはこんな感じ。

1 * 128 * 160の要素がDoubleなMultiArrayである、と。

単眼カメラで、Core MLのFCRN-DepthPredictionモデルを使ってデプス推定したもの。かなり良いのでは。ちなみに新APIは使ってないのでiOS 11+で動く。 pic.twitter.com/nDSVdLnPue

— Shuichi Tsutsumi (@shu223) July 9, 2019

このMultiArrayの扱いに関して、諸々のメモ。

MLMultiArrayのshapeのどの次元が幅で、高さで、チャンネルなのか