onehot_encodeを使用している間違った入力形式のエラー？

Question

私はmx.nd.onehot_encode関数を使用しようとしていますが、これは簡単なはずですが、解析が困難なエラーが発生しています。ここで私が試している使用例です。私はこれが1に設定のみ最初の要素と（M0と同じアドレスで）15 DIMベクトルを返すことを期待

m0 = mx.nd.zeros(15) 
mx.nd.onehot_encode(mx.nd.array([0]), m0) 

代わりに、私はエラーを取得する：

src/ndarray/./ndarray_function.h:73: Check failed: index.ndim() == 1 && proptype.ndim() == 2 OneHotEncode only support 1d index. 

どちらndarray寸法であります2、なぜこのエラーが発生するのですか？私が使用しなければならない他の入力フォーマットがありますか？

Answer 1

mxnet.ndarray.onehot_encodeは、対象のndarrayに明示的に[1、X]という形を必要とするようです。

私が試した：

m0 = mx.nd.zeros((1, 15)) 
mx.nd.onehot_encode(mx.nd.array([0]), m0) 

それはエラーを報告しません。