   y_out->at(0,0,0,0) = 0;
   y_out->at(0,0,0,1) = 1;
   y_out->at(0,0,0,2) = 0;
-
+  y_label->at(0,0,0,0) = 1;
 
   in_tensor->at(1, 0, 0, 0) = 0;
   in_tensor->at(1, 0, 0, 1) = 0;

  y_out->at(3,0,0,0) = 1;
  y_out->at(3,0,0,1) = 0;
  y_out->at(3,0,0,2) = 0;
}

void test_fc_bp_gpu() {

  h1.GetTopsDims({in_dims}, {h1_dims});
  size_t out_dims[4];
  out.GetTopsDims({h1_dims}, {out_dims});
  size_t label_dims[4];
  label_dims[0] = out_dims[0]; label_dims[1] = out_dims[1]; label_dims[2] = out_dims[2]; label_dims[3] = 1; 
  size_t softmax_out_dims[4];
  softmax_layer.GetTopsDims({out_dims}, {softmax_out_dims});
  size_t ce_out_dims[4];
