fix infer net word

PaddleRec/tdm/tdm_demo/infer_network.py

@@ -104,6 +104,11 @@ class TdmInferNet(object):
         )
 
         for layer_idx in range(self.first_layer_idx, self.max_layers):
+
+            current_layer_node = fluid.layers.reshape(
+                current_layer_node, [self.batch_size, -1])
+            current_layer_child_mask = fluid.layer.reshape(
+                current_layer_child_mask, [self.batch_size, -1])
             current_layer_node_num = current_layer_node.shape[1]
 
             node_emb = fluid.embedding(
@@ -142,7 +147,7 @@ class TdmInferNet(object):
 
             if layer_idx < self.max_layers - 1:
                 current_layer_node, current_layer_child_mask = \
-                    fluid.contribs.layers.tdm_child(x=top_node,
+                    fluid.contrib.layers.tdm_child(x=top_node,
                                                    node_nums=self.node_nums,
                                                    child_nums=self.child_nums,
                                                    param_attr=fluid.ParamAttr(name="TDM_Tree_Info"), dtype='int64')