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