[6章] 01_adding_task_rnn_encoder_decoder_tensorflow.py の検証時のデコーダーでの隠れ層の計算について

[hotsuyuki]

わかりやすい参考書を執筆して頂きありがとうございます，大変勉強になります．
01_adding_task_rnn_encoder_decoder_tensorflow.py の検証時のデコーダーでの隠れ層の計算について質問があります．
https://github.com/yusugomori/deeplearning-tensorflow-keras/blob/master/6/tensorflow/01_adding_task_rnn_encoder_decoder_tensorflow.py#L58

one hot 配列を作る際の引数 depth は正しくは以下のような値になると思うのですが，いかがでしょうか？

out = tf.one_hot(tf.argmax(out, -1), depth=output_digits) ### out.shape ---> (None, output_digits) ---> (None, 4)

↓

out = tf.one_hot(tf.argmax(out, -1), depth=n_out) ### out.shape ---> (None, n_out) ---> (None, 12)

※ 学習時のデコーダーでの隠れ層の計算に用いている配列 y[:, t-1, :] は確認してみたところ， y[:, t-1, :].shape ---> (None, 12) でした．
https://github.com/yusugomori/deeplearning-tensorflow-keras/blob/master/6/tensorflow/01_adding_task_rnn_encoder_decoder_tensorflow.py#L52