sequence_softmax

    API属性:声明式编程(静态图)专用API

    ( input, use_cudnn=False, name=None ) [源代码]

    注解

    该OP的输入只能是LoDTensor,如果要处理的输入是Tensor类型,请使用

    该OP根据LoD信息将输入的第0维度进行划分,在划分的每一个区间内部进行运算。

    对第i个区间内的元素的计算公式如下:

    输入Tensor的维度可为

    或者

    sequence_softmax - 图3

    ,推荐使用

    例如,对有6个样本的batch,每个样本的长度为3,2,4,1,2,3,其lod信息为[[0, 3, 5, 9, 10, 12, 15]],根据lod信息将第0维度划分为6份,在

    sequence_softmax - 图5

    中进行softmax运算。

    根据区间计算softmax之后的LoDTensor,其维度与input的维度一致,数据类型与input的数据类型一致。

    Variable

    1. import paddle.fluid as fluid
    2. x = fluid.data(name='x', shape=[7, 1],
    3. x_sequence_softmax = fluid.layers.sequence_softmax(input=x)
    4. dtype='float32', lod_level=1)