MultiLabelMarginLoss — PyTorch 1.10.1 documentation
pytorch.org › torchMultiLabelMarginLoss¶ class torch.nn. MultiLabelMarginLoss (size_average = None, reduce = None, reduction = 'mean') [source] ¶ Creates a criterion that optimizes a multi-class multi-classification hinge loss (margin-based loss) between input x x x (a 2D mini-batch Tensor) and output y y y (which is a 2D Tensor of target class indices). For ...
MultiLabelSoftMarginLoss — PyTorch 1.10.1 documentation
pytorch.org › docs › stableLearn about PyTorch’s features and capabilities. Community. Join the PyTorch developer community to contribute, learn, and get your questions answered. Developer Resources. Find resources and get questions answered. Forums. A place to discuss PyTorch code, issues, install, research. Models (Beta) Discover, publish, and reuse pre-trained models