Skip to content

Commit d16eb73

Browse files
authored
Fix docstring of connector _mlp_transform
Fix #185
2 parents a6f9615 + bcc80ac commit d16eb73

File tree

2 files changed

+5
-6
lines changed

2 files changed

+5
-6
lines changed

CHANGELOG.md

+2
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,8 @@
77

88
### Fixes
99

10+
* Fix docstring of connector `_mlp_transform`. ([#192](https://github.com/asyml/texar/pull/192))
11+
1012
## [v0.2.2](https://github.com/asyml/texar/releases/tag/v0.2.2) (2019-08-05)
1113

1214
### New features

texar/modules/connectors/connectors.py

+3-6
Original file line numberDiff line numberDiff line change
@@ -75,12 +75,9 @@ def _mlp_transform(inputs, output_size, activation_fn=tf.identity):
7575
with specified size.
7676
7777
Args:
78-
inputs: A Tensor of shape `[batch_size, ...]` (i.e., batch-major), or a
79-
(nested) tuple of such elements. A Tensor or a (nested) tuple of
80-
Tensors with shape `[max_time, batch_size, ...]` (i.e., time-major)
81-
can be transposed to batch-major using
82-
:func:`~texar.utils.transpose_batch_time` prior to this
83-
function.
78+
inputs: A Tensor of shape `[batch_size, d1, ..., dn]`, or a (nested)
79+
tuple of such elements. The dimensions `d1, ..., dn` will be flatten
80+
and transformed by a dense layer.
8481
output_size: Can be an Integer, a TensorShape, or a (nested) tuple of
8582
Integers or TensorShape.
8683
activation_fn: Activation function applied to the output.

0 commit comments

Comments
 (0)