Skip to content

Commit

Permalink
minor update rnn encoder
Browse files Browse the repository at this point in the history
Former-commit-id: e2ce7dc
  • Loading branch information
ZhitingHu committed Jun 5, 2018
1 parent 3497526 commit cfa4f81
Showing 1 changed file with 2 additions and 4 deletions.
6 changes: 2 additions & 4 deletions texar/modules/encoders/rnn_encoders.py
Original file line number Diff line number Diff line change
Expand Up @@ -79,8 +79,7 @@ def _build_dense_output_layer(hparams):

return dense_layers

def _forward_single_output_layer(inputs, input_size, output_layer,
flatten_inputs=True):
def _forward_single_output_layer(inputs, input_size, output_layer):
"""Forwards the input through a single output layer.
Args:
Expand All @@ -91,8 +90,7 @@ def _forward_single_output_layer(inputs, input_size, output_layer,
"""
dim = np.prod(input_size)
inputs_flat = inputs
if flatten_inputs:
inputs_flat = tf.reshape(inputs_flat, [-1, dim])
inputs_flat = tf.reshape(inputs_flat, [-1, dim])
# Feed to the layer
output_flat = output_layer(inputs_flat)
output_size = output_layer.compute_output_shape([1, dim]).as_list()[1:]
Expand Down

0 comments on commit cfa4f81

Please sign in to comment.