当前位置: 首页>>代码示例 >>用法及示例精选 >>正文


Python mxnet.ndarray.SequenceLast用法及代码示例


用法:

mxnet.ndarray.SequenceLast(data=None, sequence_length=None, use_sequence_length=_Null, axis=_Null, out=None, name=None, **kwargs)

参数

  • data(NDArray) - 形式为 [max_sequence_length, batch_size, other_feature_dims] 的 n 维输入数组,其中 n>2
  • sequence_length(NDArray) - [batch_size] 形式的序列长度向量
  • use_sequence_length(boolean, optional, default=0) - 如果设置为 true,则该层接受一个额外的输入参数sequence_length指定可变长度序列
  • axis(int, optional, default='0') - 序列轴。当前仅支持 0 和 1 的值。
  • out(NDArray, optional) - 输出 NDArray 来保存结果。

返回

out- 此函数的输出。

返回类型

NDArray 或 NDArray 列表

获取序列的最后一个元素。

此函数采用 [max_sequence_length, batch_size, other_feature_dims] 形式的 n 维输入数组,并返回 [batch_size, other_feature_dims] 形式的 (n-1) 维数组。

参数sequence_length 用于处理可变长度序列。 sequence_length 应该是维度为 [batch_size] 的正整数的输入数组。要使用此参数,请将 use_sequence_length 设置为 True ,否则将假定批处理中的每个示例都具有最大序列长度。

注意

或者,您也可以使用take 运算符。

例子:

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.]]

相关用法


注:本文由纯净天空筛选整理自apache.org大神的英文原创作品 mxnet.ndarray.SequenceLast。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。