class_docstr: Output layer for word tagging models. It supports `CrossEntropyLoss` per word.
  
  Args:
      loss_fn (CrossEntropyLoss): Cross-entropy loss component. Defaults to None.
  
  Attributes:
      loss_fn: Cross-entropy loss component.
class_docstr_tok: ["Output","layer","for","word","tagging","models",".","It","supports","`","CrossEntropyLoss","`","per","word",".","Args",":","loss_fn","(","CrossEntropyLoss",")",":","Cross-entropy","loss","component",".","Defaults","to","None",".","Attributes",":","loss_fn",":","Cross-entropy","loss","component","."]
docstr: Exports the word tagging output layer to Caffe2.
docstr_tok: ["Exports","the","word","tagging","output","layer","to","Caffe2","."]
code: def export_to_caffe2(
      self,
      workspace: core.workspace,
      init_net: core.Net,
      predict_net: core.Net,
      model_out: torch.Tensor,
      output_name: str,
  ) -> List[core.BlobReference]:
      """Exports the word tagging output layer to Caffe2."""
      probability_out = predict_net.Softmax(
          output_name, axis=model_out.dim() - 1
      )
      return OutputLayerUtils.gen_additional_blobs(
          predict_net,
          probability_out,
          model_out,
          output_name,
          self.target_names,
      )
  
imports: from typing import Any, Dict, List, Optional, Tuple, Union
  import torch
  import torch.jit
  import torch.nn
  import torch.nn.functional
  from caffe2.python import core
  from pytext.common import Padding
  from pytext.config.component import create_loss
  from pytext.config.serialize import MissingValueError
  from pytext.data.utils import Vocabulary
  from pytext.loss import AUCPRHingeLoss, BinaryCrossEntropyLoss, CrossEntropyLoss, KLDivergenceBCELoss, KLDivergenceCELoss, LabelSmoothedCrossEntropyLoss, StructuredLoss, StructuredMarginLoss
  from pytext.models.crf import CRF
  from pytext.utils.label import get_label_weights
  from pytext.utils.usage import log_class_usage
  from .output_layer_base import OutputLayerBase
  from .utils import OutputLayerUtils
label_desc: Use image node_id 4 for calling the WordTaggingOutputLayer obj's underlying member method code with example usage: obj.export_to_caffe2(workspace, init_net, predict_net, model_out, output_name) and returns: OutputLayerUtils