Function torch::nn::functional::one_hot

• Defined in File embedding.h

Function Documentation

inline Tensor torch::nn::functional::one_hot(const Tensor &tensor, int64_t num_classes = -1)