def accuracy(out, [yb](https://docs.pytorch.org/docs/stable/tensors.html#torch.Tensor "torch.Tensor")):
    [preds](https://docs.pytorch.org/docs/stable/tensors.html#torch.Tensor "torch.Tensor") = [torch.argmax](https://docs.pytorch.org/docs/stable/generated/torch.argmax.html#torch.argmax "torch.argmax")(out, dim=1)
    return ([preds](https://docs.pytorch.org/docs/stable/tensors.html#torch.Tensor "torch.Tensor") == [yb](https://docs.pytorch.org/docs/stable/tensors.html#torch.Tensor "torch.Tensor")).float().mean()