org.bytedeco.tensorflow

Class OneHot

java.lang.Object

org.bytedeco.javacpp.Pointer

org.bytedeco.tensorflow.OneHot

All Implemented Interfaces:

AutoCloseable

@Namespace(value="tensorflow::ops")
 @NoOffset
 @Properties(inherit=tensorflow.class)
public class OneHot
extends Pointer

Returns a one-hot tensor. The locations represented by indices in indices take value on_value, while all other locations take value off_value. If the input indices is rank N, the output will have rank N+1, The new axis is created at dimension axis (default: the new axis is appended at the end). If indices is a scalar the output shape will be a vector of length depth. If indices is a vector of length features, the output shape will be:

    features x depth if axis == -1
    depth x features if axis == 0
  

If indices is a matrix (batch) with shape [batch, features], the output shape will be:

    batch x features x depth if axis == -1
    batch x depth x features if axis == 1
    depth x batch x features if axis == 0
  

Examples ========= Suppose that

    indices = [0, 2, -1, 1]
    depth = 3
    on_value = 5.0
    off_value = 0.0
    axis = -1
  

Then output is [4 x 3]:

  output =
    [5.0 0.0 0.0]  // one_hot(0)
    [0.0 0.0 5.0]  // one_hot(2)
    [0.0 0.0 0.0]  // one_hot(-1)
    [0.0 5.0 0.0]  // one_hot(1)
  

Suppose that

    indices = [0, 2, -1, 1]
    depth = 3
    on_value = 0.0
    off_value = 3.0
    axis = 0
  

Then output is [3 x 4]:

  output =
    [0.0 3.0 3.0 3.0]
    [3.0 3.0 3.0 0.0]
    [3.0 3.0 3.0 3.0]
    [3.0 0.0 3.0 3.0]
  //  ^                one_hot(0)
  //      ^            one_hot(2)
  //          ^        one_hot(-1)
  //              ^    one_hot(1)
  

Suppose that

    indices = [[0, 2], [1, -1]]
    depth = 3
    on_value = 1.0
    off_value = 0.0
    axis = -1
  

Then output is [2 x 2 x 3]:

  output =
    [
      [1.0, 0.0, 0.0]  // one_hot(0)
      [0.0, 0.0, 1.0]  // one_hot(2)
    ][
      [0.0, 1.0, 0.0]  // one_hot(1)
      [0.0, 0.0, 0.0]  // one_hot(-1)
    ]
  

Arguments: * scope: A Scope object * indices: A tensor of indices. * depth: A scalar defining the depth of the one hot dimension. * on_value: A scalar defining the value to fill in output when indices[j] = i. * off_value: A scalar defining the value to fill in output when indices[j] != i. Optional attributes (see Attrs): * axis: The axis to fill (default: -1, a new inner-most axis). Returns: * Output: The one-hot tensor.

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
static class	OneHot.Attrs Optional attribute setters for OneHot

Nested classes/interfaces inherited from class org.bytedeco.javacpp.Pointer

Pointer.CustomDeallocator, Pointer.Deallocator, Pointer.NativeDeallocator, Pointer.ReferenceCounter

Field Summary

Fields inherited from class org.bytedeco.javacpp.Pointer

address, capacity, limit, position

Constructor Summary

Constructors

Constructor and Description
OneHot(Pointer p) Pointer cast constructor.
OneHot(Scope scope, Input indices, Input depth, Input on_value, Input off_value)
OneHot(Scope scope, Input indices, Input depth, Input on_value, Input off_value, OneHot.Attrs attrs)

Method Summary

Modifier and Type	Method and Description
Input	asInput()
Output	asOutput()
static OneHot.Attrs	Axis(long x)
Node	node()
Operation	operation()
OneHot	operation(Operation setter)
Output	output()
OneHot	output(Output setter)

Methods inherited from class org.bytedeco.javacpp.Pointer

address, asBuffer, asByteBuffer, availablePhysicalBytes, calloc, capacity, capacity, close, deallocate, deallocate, deallocateReferences, deallocator, deallocator, equals, fill, formatBytes, free, getDirectBufferAddress, getPointer, getPointer, getPointer, getPointer, hashCode, interruptDeallocatorThread, isNull, isNull, limit, limit, malloc, maxBytes, maxPhysicalBytes, memchr, memcmp, memcpy, memmove, memset, offsetAddress, offsetof, offsetof, parseBytes, physicalBytes, physicalBytesInaccurate, position, position, put, realloc, referenceCount, releaseReference, retainReference, setNull, sizeof, sizeof, toString, totalBytes, totalCount, totalPhysicalBytes, withDeallocator, zero

Methods inherited from class java.lang.Object

clone, finalize, getClass, notify, notifyAll, wait, wait, wait

Constructor Detail

OneHot

public OneHot(Pointer p)

Pointer cast constructor. Invokes Pointer(Pointer).

OneHot

public OneHot(@Const @ByRef
              Scope scope,
              @ByVal
              Input indices,
              @ByVal
              Input depth,
              @ByVal
              Input on_value,
              @ByVal
              Input off_value)

OneHot

public OneHot(@Const @ByRef
              Scope scope,
              @ByVal
              Input indices,
              @ByVal
              Input depth,
              @ByVal
              Input on_value,
              @ByVal
              Input off_value,
              @Const @ByRef
              OneHot.Attrs attrs)

Method Detail

asOutput

@ByVal
 @Name(value="operator tensorflow::Output")
public Output asOutput()

asInput

@ByVal
 @Name(value="operator tensorflow::Input")
public Input asInput()

node

public Node node()

Axis

@ByVal
public static OneHot.Attrs Axis(@Cast(value="tensorflow::int64")
                                        long x)

operation

@ByRef
public Operation operation()

operation

public OneHot operation(Operation setter)

output

@ByRef
public Output output()

output

public OneHot output(Output setter)