Contents

oneHot(withIndicesTensor:depth:axis:dataType:onValue:offValue:name:)

Creates a oneHot operation and returns the result tensor.

Declaration

func oneHot(withIndicesTensor indicesTensor: MPSGraphTensor, depth: Int, axis: Int, dataType: MPSDataType, onValue: Double, offValue: Double, name: String?) -> MPSGraphTensor

Parameters

  • indicesTensor:

    Tensor of indices for on values

  • depth:

    Depth of the oneHot vector along the axis

  • axis:

    The axis to insert the new oneHot vector at. Defaults to -1, the minor axis

  • dataType:

    MPSDataType of the result tensor Defaults to MPSDataTypeFloat

  • onValue:

    The value for indices designated by the indicesTensor. This value must match the specified data type. Defaults to 1.0f

  • offValue:

    The value for indices not designated by the indicesTensor. This value must match the specified data type. Defaults to 0.0f

  • name:

    Name for the operation

Return Value

A valid MPSGraphTensor object.

Discussion

Creates a tensor of rank equal to the indicesTensor rank + 1. Inserts a new axis at the axis specified, or the minor axis if axis is -1. The values at the indices in the indicesTensor will have the onValue, and all other values will be set to the offValue.