blob: 01344216a8f3023ce4a48ce8a4f4ddc5f338fd15 [file] [log] [blame]
``mx.nd.SequenceLast``
============================================
Description
----------------------
Takes the last element of a sequence.
This function takes an n-dimensional input array of the form
[max_sequence_length, batch_size, other_feature_dims] and returns a (n-1)-dimensional array
of the form [batch_size, other_feature_dims].
Parameter `sequence_length` is used to handle variable-length sequences. `sequence_length` should be
an input array of positive ints of dimension [batch_size]. To use this parameter,
set `use_sequence_length` to `True`, otherwise each example in the batch is assumed
to have the max sequence length.
.. note:: Alternatively, you can also use `take` operator.
**Example**::
x = [[[ 1., 2., 3.],
[ 4., 5., 6.],
[ 7., 8., 9.]],
[[ 10., 11., 12.],
[ 13., 14., 15.],
[ 16., 17., 18.]],
[[ 19., 20., 21.],
[ 22., 23., 24.],
[ 25., 26., 27.]]]
// returns last sequence when sequence_length parameter is not used
SequenceLast(x) = [[ 19., 20., 21.],
[ 22., 23., 24.],
[ 25., 26., 27.]]
// sequence_length is used
SequenceLast(x, sequence_length=[1,1,1], use_sequence_length=True) =
[[ 1., 2., 3.],
[ 4., 5., 6.],
[ 7., 8., 9.]]
// sequence_length is used
SequenceLast(x, sequence_length=[1,2,3], use_sequence_length=True) =
[[ 1., 2., 3.],
[ 13., 14., 15.],
[ 25., 26., 27.]]
Arguments
------------------
+----------------------------------------+------------------------------------------------------------+
| Argument | Description |
+========================================+============================================================+
| ``data`` | NDArray-or-Symbol |
| | n-dimensional input array of the form |
| | [max_sequence_length, batch_size, other_feature_dims] |
| | where |
| | n>2 |
+----------------------------------------+------------------------------------------------------------+
| ``sequence.length`` | NDArray-or-Symbol |
| | vector of sequence lengths of the form [batch_size] |
+----------------------------------------+------------------------------------------------------------+
| ``use.sequence.length`` | boolean, optional, default=0. |
| | |
| | If set to true, this layer takes in an extra input |
| | parameter `sequence_length` to specify variable length |
| | sequence |
+----------------------------------------+------------------------------------------------------------+
| ``axis`` | int, optional, default='0'. |
| | |
| | The sequence axis. Only values of 0 and 1 are currently |
| | supported. |
+----------------------------------------+------------------------------------------------------------+
Value
----------
``out`` The result mx.ndarray
Link to Source Code: http://github.com/apache/incubator-mxnet/blob/1.6.0/src/operator/sequence_last.cc#L106