fix expand layer

PaddleRec/tdm/tdm_demo/train_network.py

@@ -199,6 +199,10 @@ class DnnLayerClassifierNet(object):
     def _expand_layer(self, input_layer, node, layer_idx):
         input_layer_unsequeeze = fluid.layers.unsqueeze(
             input=input_layer, axes=[1])
+        if self.is_test:
+            input_layer_expand = fluid.layers.expand(
+                input_layer_unsequeeze, expand_times=[1, node.shape[1], 1])
+        else:
             input_layer_expand = fluid.layers.expand(
                 input_layer_unsequeeze, expand_times=[1, node[layer_idx].shape[1], 1])
         return input_layer_expand