File tree Expand file tree Collapse file tree 1 file changed +4
-1
lines changed Expand file tree Collapse file tree 1 file changed +4
-1
lines changed Original file line number Diff line number Diff line change @@ -185,6 +185,7 @@ def data(name,
185
185
shape ,
186
186
append_batch_size = True ,
187
187
dtype = 'float32' ,
188
+ lod_level = 0 ,
188
189
type = core .VarDesc .VarType .LOD_TENSOR ,
189
190
main_program = None ,
190
191
startup_program = None ,
@@ -198,6 +199,7 @@ def data(name,
198
199
append_batch_size: Whether or not to append the data as a batch.
199
200
dtype: The type of data : float32, float_16, int etc
200
201
type: The output type. By default it is LOD_TENSOR.
202
+ lod_level(int): The LoD Level. 0 means the input data is not a sequence.
201
203
main_program: Name of the main program that calls this
202
204
startup_program: Name of the startup program
203
205
stop_gradient: A boolean that mentions whether gradient should flow.
@@ -228,7 +230,8 @@ def data(name,
228
230
shape = shape ,
229
231
dtype = dtype ,
230
232
type = type ,
231
- stop_gradient = stop_gradient )
233
+ stop_gradient = stop_gradient ,
234
+ lod_level = lod_level )
232
235
233
236
234
237
def create_tensor (dtype , name = None , main_program = None , startup_program = None ):
You can’t perform that action at this time.
0 commit comments