float型のndarrayのreshapeをしてcsvに保存したい

Question

機械学習の学習用データのテンソルをcsvに保存しようとしています．
csvに保存する関係で，3次元のテンソルを2次元にreshapeして保存しようとしたのですが，reshapeはint型のndarrayでしか出来ないらしく，どうすれば保存できるかわからず，困っています．

以下のコードの#3次元以上の配列は~以降でreshapeを行なっていますが，TypeError: only integer scalar arrays can be converted to a scalar index
とエラーが出てしまいました．

---

試した事

• float型からint型にキャストしてreshapeして再びfloat型に直して保存

結果

• 同様のエラーが発生

# ベクトル化した学習データを用意
        embeded_sequences = np.zeros((sequence.shape[0], maxlen, dim))  # 空のndarrayを生成([データ数*max_len*dim])
        if os.path.exists(training_dir):
            # トレーニングデータ読み込み
            embeded_sequences = np.loadtxt(training_dir, delimiter=',')
        else:
            # トレーニングデータの作成と保存
            print('make training data !!')
            for i, row in enumerate(sequence):
                for j, data in enumerate(row):
                    embedding_vector = embedding_metrix[data]
                    embeded_sequences[i][j] = embedding_vector

            # 3次元以上の配列はcsvに保存できないので二次元[データ数*(max_len*dim)]に整形
            embeded_sequences.reshape([embeded_sequences.shape[0], embeded_sequences.shape[1] * embeded_sequences[2]])
            np.savetxt(fname=training_dir, X=embeded_sequences, delimiter=',')
            print('save training data !!')

Answer 1

embeded_sequences[2]が間違っているだけだと思います。

-            embeded_sequences.reshape([embeded_sequences.shape[0], embeded_sequences.shape[1] * embeded_sequences[2]])
+            embeded_sequences.reshape([embeded_sequences.shape[0], embeded_sequences.shape[1] * embeded_sequences.shape[2]])

Comments

[iforin]

回答ありがとうございます！見逃していました！
ご指摘の通り修正した結果，質問に記述したエラーは出なくなりましたが，なぜか３次元のままになってしまい，csvへの保存ができませんでした．
もしこの問題に関して，何か解決方法がありましたら教えていただけないでしょうか？

[hayataka2049]

reshapeはもとの配列はそのままにしてreshapeした結果を返すメソッドなので、結果を適当に代入しないと無意味ですね。
embeded_sequences = embeded_sequences.reshape( # 省略
とかで。

[iforin]

ありがとうございます！無事解決できました！