rising.ops
Provides Operators working on single tensors.
On Tensors
torch_one_hot
-
rising.ops.tensor.torch_one_hot(target, num_classes=None)[source][source]
Compute one hot encoding of input tensor
- Parameters
target (Tensor
) – tensor to be converted
num_classes (Optional
[int
]) – number of classes. If num_classes
is None,
the maximum of target is used
- Returns
one hot encoded tensor
- Return type
torch.Tensor
np_one_hot
-
rising.ops.tensor.np_one_hot(target, num_classes=None)[source][source]
Compute one hot encoding of input array
- Parameters
-
- Returns
one hot encoded array
- Return type
numpy.ndarray