Fix issue caused by argmin/argmax output shape

[mingmingtasd]

No description provided.

[fdwr]

Propose the following, just checking original_output_tensor_desc and calling CreateNodeOutput once, rather than calling CreateNodeOutput once and then again, overwriting a previous call.

...
  auto output_data_type = GetTensorDataType(output_desc->data_type);
  TensorDesc output_tensor_desc(output_data_type, output_dimensions);
  TensorDesc original_output_tensor_desc(output_data_type,  //<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<
                                         output_desc->dimensions);

  // DML accepts multiple axes. So pass the single index along.
  std::array<uint32_t, 1> axes = {axis};

...
  Node node = graph_desc_builder_->CreateOperatorNode(dml_operator_type,
                                                      &operator_desc);

  graph_desc_builder_->Connect({input_node}, {node});

  auto node_output = graph_desc_builder_->CreateNodeOutput(
      node, 0, std::move(original_output_tensor_desc)); //<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<
  node_output_map_[output_index] = std::move(node_output);

(no need to keep_dimensions down the stack, since the original dimensions already adjusted that)

[mingmingtasd]

Thanks, updated, PTAL! @fdwr