I fixed this issue as well by following @jessylol suggestion. All you have to do is declare the stateIn param like this let array = Array(repeating: 0, count: 400) let stateIn = try MLMultiArray(array)