Class EmbeddingActivations

java.lang.Object

org.tensorflow.op.RawOp

org.tensorflow.op.tpu.EmbeddingActivations

All Implemented Interfaces:

Shaped, Op, Operand<TFloat32>

@Operator(group="tpu")
public final class EmbeddingActivations
extends RawOp
implements Operand<TFloat32>

An op enabling differentiation of TPU Embeddings. This op simply returns its first input, which is assumed to have been sliced from the Tensors returned by TPUEmbeddingDequeueActivations. The presence of this op, and its first argument being a trainable Variable, enables automatic differentiation of graphs containing embeddings via the TPU Embedding Python libraries.

Nested Class Summary

Nested Classes

Modifier and Type	Class	Description
static class	EmbeddingActivations.Inputs

Field Summary

Fields

Modifier and Type	Field	Description
static final String	OP_NAME	The name of this op, as known by TensorFlow core engine

Fields inherited from class RawOp

operation

Constructor Summary

Constructors

Constructor	Description
EmbeddingActivations(Operation operation)

Method Summary

Modifier and Type	Method	Description
Output<TFloat32>	asOutput()	Returns the symbolic handle of the tensor.
static EmbeddingActivations	create(Scope scope, Operand<TFloat32> embeddingVariable, Operand<TFloat32> slicedActivations, Long tableId, Long lookupId)	Factory method to create a class wrapping a new TPUEmbeddingActivations operation.
Output<TFloat32>	output()	Gets output.

Methods inherited from class RawOp

equals, hashCode, op, toString

Methods inherited from class Object

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

Methods inherited from interface Op

env, op

Methods inherited from interface Operand

asTensor, shape, type

Methods inherited from interface Shaped

rank, size

Field Details

OP_NAME

public static final String OP_NAME

The name of this op, as known by TensorFlow core engine

See Also:

• Constant Field Values

Constructor Details

EmbeddingActivations

public EmbeddingActivations(Operation operation)

Method Details

create

@Endpoint(describeByClass=true)
public static EmbeddingActivations create(Scope scope,
 Operand<TFloat32> embeddingVariable,
 Operand<TFloat32> slicedActivations,
 Long tableId,
 Long lookupId)

Factory method to create a class wrapping a new TPUEmbeddingActivations operation.

Parameters:

scope - current scope

embeddingVariable - A trainable variable, enabling optimizers to find this op.

slicedActivations - The embedding activations Tensor to return.

tableId - The id of the table in the embedding layer configuration from which these activations were computed.

lookupId - Identifier of the set of embedding indices which produced these activations.

Returns:

a new instance of EmbeddingActivations

output

public Output<TFloat32> output()

Gets output.

Returns:

output.

asOutput

public Output<TFloat32> asOutput()

Description copied from interface: Operand

Returns the symbolic handle of the tensor.

Inputs to TensorFlow operations are outputs of another TensorFlow operation. This method is used to obtain a symbolic handle that represents the computation of the input.

Specified by:

asOutput in interface Operand<TFloat32>

See Also:

• OperationBuilder.addInput(Output)