org.tensorflow.op.core

Class TPUEmbeddingActivations

java.lang.Object

org.tensorflow.op.PrimitiveOp

org.tensorflow.op.core.TPUEmbeddingActivations

All Implemented Interfaces:

Op, Operand<Float>

public final class TPUEmbeddingActivations
extends PrimitiveOp
implements Operand<Float>

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.

Field Summary

Fields inherited from class org.tensorflow.op.PrimitiveOp

operation

Method Summary

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

Methods inherited from class org.tensorflow.op.PrimitiveOp

equals, hashCode, op, toString

Methods inherited from class java.lang.Object

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

Method Detail

create

public static TPUEmbeddingActivations create(Scope scope,
                                             Operand<Float> embeddingVariable,
                                             Operand<Float> 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 TPUEmbeddingActivations

output

public Output<Float> output()

asOutput

public Output<Float> asOutput()

Description copied from interface: Operand

Returns the symbolic handle of a 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<Float>

See Also:

OperationBuilder.addInput(Output)