code: def CrossEntropyFlat(*args, axis:int=-1, **kwargs):
      "Same as `nn.CrossEntropyLoss`, but flattens input and target."
      return FlattenedLoss(nn.CrossEntropyLoss, *args, axis=axis, **kwargsf CrossEntropyFlat(*args, axis:int=-1, **kwargs):
      "Same as `nn.CrossEntropyLoss`, but flattens input and target."
      return FlattenedLoss(nn.CrossEntropyLoss, *args, axis=axis, **kwargs)
docs: Same as `nn.CrossEntropyLoss`, but flattens input and target.