Returns:
            The state probabilities

def probabilities(self) -> bk.BKTensor:
        """
        Returns:
            The state probabilities
        """
        value = bk.absolute(self.tensor)
        return value * value