Skip to content

Commit

Permalink
added output layers for bi-rnn encoder
Browse files Browse the repository at this point in the history
Former-commit-id: 419db5d
  • Loading branch information
ZhitingHu committed May 30, 2018
1 parent ebec93d commit 77c08db
Show file tree
Hide file tree
Showing 4 changed files with 375 additions and 64 deletions.
8 changes: 4 additions & 4 deletions texar/modules/connectors/connectors.py
Original file line number Diff line number Diff line change
Expand Up @@ -414,9 +414,9 @@ class which inherits
to generate. `None` is required in training stage.
Returns:
output: If `num_samples`==None, returns a Tensor of shape
`[batch_size x output_size]`, else returns a Tensor of shape
`[num_samples x output_size]`. `num_samples` should be specified
output: If `num_samples`==None, returns a Tensor of shape
`[batch_size x output_size]`, else returns a Tensor of shape
`[num_samples x output_size]`. `num_samples` should be specified
if not in training stage.
latent_z: The latent sampled z
Expand Down Expand Up @@ -644,7 +644,7 @@ def _build(self, connector_inputs, transform=True):
Args:
connector_inputs: a list of connector states
transform: If ``True``, then the output are automatically
transform: If `True`, then the output are automatically
transformed to match :attr:`output_size`.
Returns:
Expand Down
2 changes: 1 addition & 1 deletion texar/modules/decoders/rnn_decoders.py
Original file line number Diff line number Diff line change
Expand Up @@ -355,7 +355,7 @@ def default_hparams():
Here:
"attention" : dict
A dictionary of attention hyperparameters, which includes:
Attention hyperparameters, which includes:
"type" : str
Name or full path to the attention class which can be
Expand Down

0 comments on commit 77c08db

Please sign in to comment.