tensordict.nn.rand\_one\_hot ============================ .. currentmodule:: tensordict.nn .. autofunction:: rand_one_hot