TensorFlowでLSTMセルを初期化する方法は？

Question

TF v1.0で次のコードを実行しようとしていて、エラーが発生します。私はLSTMを作成し、その後、出力を計算するためにlstm_cellするためにそれを渡すために、状態変数を定義しますが、状態は初期化できませんしている。ここで

lstm_cell = tf.contrib.rnn.BasicLSTMCell(10) 
# Initial state of the LSTM memory. 
state = tf.zeros([20, lstm_cell.state_size]) 
outputs, states = lstm_cell(x , state) 

それトレースバック：

ValueError        Traceback (most recent call last) 
<ipython-input-82-4a23eee1acf4> in <module>() 
     1 lstm_cell = tf.contrib.rnn.BasicLSTMCell(10) 
     2 # Initial state of the LSTM memory. 
----> 3 state = tf.zeros([20, lstm_cell.state_size]) 
     4 outputs, states = lstm_cell(x , state) 

/Users/Saeed/anaconda/lib/python2.7/site-packages/tensorflow/python/ops/array_ops.pyc in zeros(shape, dtype, name) 
    1370  output = constant(zero, shape=shape, dtype=dtype, name=name) 
    1371  except (TypeError, ValueError): 
-> 1372  shape = ops.convert_to_tensor(shape, dtype=dtypes.int32, name="shape") 
    1373  output = fill(shape, constant(zero, dtype=dtype), name=name) 
    1374 assert output.dtype.base_dtype == dtype 

/Users/Saeed/anaconda/lib/python2.7/site-packages/tensorflow/python/framework/ops.pyc in convert_to_tensor(value, dtype, name, preferred_dtype) 
    649  name=name, 
    650  preferred_dtype=preferred_dtype, 
--> 651  as_ref=False) 
    652 
    653 

/Users/Saeed/anaconda/lib/python2.7/site-packages/tensorflow/python/framework/ops.pyc in internal_convert_to_tensor(value, dtype, name, as_ref, preferred_dtype) 
    714 
    715   if ret is None: 
--> 716   ret = conversion_func(value, dtype=dtype, name=name, as_ref=as_ref) 
    717 
    718   if ret is NotImplemented: 

/Users/Saeed/anaconda/lib/python2.7/site-packages/tensorflow/python/framework/constant_op.pyc in _constant_tensor_conversion_function(v, dtype, name, as_ref) 
    174           as_ref=False): 
    175 _ = as_ref 
--> 176 return constant(v, dtype=dtype, name=name) 
    177 
    178 

/Users/Saeed/anaconda/lib/python2.7/site-packages/tensorflow/python/framework/constant_op.pyc in constant(value, dtype, shape, name, verify_shape) 
    163 tensor_value = attr_value_pb2.AttrValue() 
    164 tensor_value.tensor.CopyFrom(
--> 165  tensor_util.make_tensor_proto(value, dtype=dtype, shape=shape, verify_shape=verify_shape)) 
    166 dtype_value = attr_value_pb2.AttrValue(type=tensor_value.tensor.dtype) 
    167 const_tensor = g.create_op(

/Users/Saeed/anaconda/lib/python2.7/site-packages/tensorflow/python/framework/tensor_util.pyc in make_tensor_proto(values, dtype, shape, verify_shape) 
    366  else: 
    367  _AssertCompatible(values, dtype) 
--> 368  nparray = np.array(values, dtype=np_dt) 
    369  # check to them. 
    370  # We need to pass in quantized values as tuples, so don't apply the shape 

ValueError: setting an array element with a sequence. 

Answer 1

このTensorflowの最新バージョンでは、デフォルトのBasicLSTMCellのstate_sizeプロパティの戻り値がLSTMStateTuple（Python Tuple）であるためです。

source codeをチェックすると、タプルの両方の要素（以前のバージョンでは同じ軸に沿って連結されていた）に同じ数の単位が返され、セル状態を初期化するときに考慮する必要があります。

state = tf.zeros([20, lstm_cell.state_size[0]*2) 

：

したがって、これはトリックを行う必要があります