Quelle devrait être la taille de l'entrée et de l'état caché dans GRUCell de tensorflow (python)?

Question

Je suis nouveau à tensorflow (1 jour d'expérience).

Je suis en train suivant petit code pour créer un simple RNN à base GRU avec une seule couche et la taille cachée de 100 comme suit:

import pickle 
import numpy as np 
import pandas as pd 
import tensorflow as tf 

# parameters 
batch_size = 50 
hidden_size = 100 

# create network graph 
input_data = tf.placeholder(tf.int32, [batch_size]) 
output_data = tf.placeholder(tf.int32, [batch_size]) 

cell = tf.nn.rnn_cell.GRUCell(hidden_size) 

initial_state = cell.zero_state(batch_size, tf.float32) 

hidden_state = initial_state 

output_of_cell, hidden_state = cell(input_data, hidden_state) 

Mais j'obtiens l'erreur suivante pour la dernière ligne (c.-à appeler à cell())

Linear is expecting 2D arguments: [[50], [50, 100]] 

Qu'est-ce que je fais mal?

Answer 1

Les entrées de l'opérateur d'appel GRUCell devraient être des tenseurs 2-D de type tf.float32. Ce qui suit devrait fonctionner:

input_data = tf.placeholder(tf.float32, [batch_size, input_size]) 

cell = tf.nn.rnn_cell.GRUCell(hidden_size) 

initial_state = cell.zero_state(batch_size, tf.float32) 

hidden_state = initial_state 

output_of_cell, hidden_state = cell(input_data, hidden_state)