mm.math.Categorical

math.Categorical(name)

Categorical distribution over integers.

Parameters:
  • probs (ndarray[Tuple[int, ...], Union[TypeVar(R, float16, float32, float64), TypeVar(C, complex64, complex128), TypeVar(Z, int16, int32, int64), TypeVar(N, uint16, uint32, uint64)]]) – The unnormalized probabilities of a set of Categorical distributions.

  • name (str) – The name prefixed to operations created by this class.

Returns:

instance of tfp.distributions.Categorical class

Return type:

tfp.distributions.Categorical