Created by: xxxsssyyy

想实现的操作： 举例: 现在有一个lodTensor lod为 [ [0, 2, 5] ] data 为 [ [1.1, 2.1, 5.2] , [5.2, 0.1, 4.3] , [0.2, 10.5, 8.3] , [1, 2.5, 3.3] , [7.1, 2.1, 5.9] , ]

想通过一个argmax操作 对axis = -1 得到的 lod 为[ [0, 2, 5] ] data 为 [[2] [0] [1] [2] [0]]

现在的代码为:

emission = fluid.layers.fc(
            size=label_dict_len,
            input=bigru_output,
            param_attr=fluid.ParamAttr(
                initializer=fluid.initializer.Uniform(
                    low=-init_bound, high=init_bound),
                regularizer=fluid.regularizer.L2DecayRegularizer(
                    regularization_coeff=1e-4))
        )
softmax_decode = fluid.layers.argmax(x=emission, axis=-1)

目前的问题：结果中softmax_decode 的lod为空的，想问一下如何解决（主要是实现我上面的需求）