t contribute to the metric calculation validate_args: bool indicating if input arguments and tensors should be validated for correctness. Set to ``False`` for faster computations. Returns: The returned shape depends on the ``average`` and ``multidim_average`` arguments: - If ``multidim_average`` is set to ``global``: - If ``average='micro'/'macro'/'weighted'``, the output will be a scalar tensor - If ``average=None/'none'``, the shape will be ``(C,)`` - If ``multidim_average`` is set to ``samplewise``: - If ``average='micro'/'macro'/'weighted'``, the shape will be ``(N,)`` - If ``average=None/'none'``, the shape will be ``(N, C)`` Example (preds is int tensor): >>> from torch import tensor >>> from torchmetrics.functional.classification import multilabel_f1_score >>> target = tensor([[0, 1, 0], [1, 0, 1]]) >>> preds = tensor([[0, 0, 1], [1, 0, 1]]) >>> multilabel_f1_score(preds, target, num_labels=3) tensor(0.5556) >>> multilabel_f1_score(preds, target, num_labels=3, average=None) tensor([1.0000, 0.0000, 0.6667]) Example (preds is float tensor): >>> from torchmetrics.functional.classification import multilabel_f1_score >>> target = tensor([[0, 1, 0], [1, 0, 1]]) >>> preds = tensor([[0.11, 0.22, 0.84], [0.73, 0.33, 0.92]]) >>> multilabel_f1_score(preds, target, num_labels=3) tensor(0.5556) >>> multilabel_f1_score(preds, target, num_labels=3, average=None) tensor([1.0000, 0.0000, 0.6667]) Example (multidim tensors): >>> from torchmetrics.functional.classification import multilabel_f1_score >>> target = tensor([[[0, 1], [1, 0], [0, 1]], [[1, 1], [0, 0], [1, 0]]]) >>> preds = tensor([[[0.59, 0.91], [0.91, 0.99], [0.63, 0.04]], ... [[0.38, 0.04], [0.86, 0.780], [0.45, 0.37]]]) >>> multilabel_f1_score(preds, target, num_labels=3, multidim_average='samplewise') tensor([0.4444, 0.0000]) >>> multilabel_f1_score(preds, target, num_labels=3, multidim_average='samplewise', average=None) tensor([[0.6667, 0.6667, 0.0000], [0.0000, 0.0000, 0.0000]]) rB