uniport.model.utils.onehot

uniport.model.utils.onehot(y, n)[source]

Make the input tensor one hot tensors

Parameters:

  • y – input tensors

  • n – number of classes

Return type:

Tensor