【TensorFlow】RNNで使用するものまとめ

【TensorFlowについて】

version: 1.2.0

import tensorflow as tf

 

【クラス、メソッド等】

■tf.nn.rnn_cell.BasicRNNCell(num_units)

num_units: 中間層のユニット数

 

使用例と説明

mid_cell = tf.nn.rnn_cell.BasicRNNCell(30)

ユニット数30のRNNの中間層がmid_cellとして定義される。

通常のRNNの中間層であり、LSTMなどではない。

 

■tf.nn.static_rnn(cell, inputs, initial_state)

cell:定義したRNNの中間層

inputs:中間層への入力データ(入力層の出力)

initial_state: