torch.utils.tensorboard — PyTorch 1.10.1 documentation
pytorch.org › docs › stableThe TensorBoard UI will let you choose the threshold interactively. Parameters. tag (string) – Data identifier. labels (torch.Tensor, numpy.array, or string/blobname) – Ground truth data. Binary label for each element. predictions (torch.Tensor, numpy.array, or string/blobname) – The probability that an element be classified as true ...