scalex.net.utils.onehot

scalex.net.utils.onehot(y, n)

Make the input tensor one hot tensors

Parameters

  • y – input tensors

  • n – number of classes

Return type

Tensor