def combined_output_dim(self, output_dims):
    """Determines the output dimension of the combined tensor.

    Args:
      output_dims: List of output dimensions of combined tensors.

    Returns:
      Output dimension of the combined tensor.
    """
Determines the output dimension of the combined tensor.

    Args:
      output_dims: List of output dimensions of combined tensors.

    Returns:
      Output dimension of the combined tensor.
    
tensor2tensor/models/neural_architecture_search/nas_model.py