public final class Ops extends Object
Op
s
Any operation wrapper found in the classpath properly annotated as an@Operator
is exposed
by this API or one of its subgroup.
Example usage:
try (Graph g = new Graph()) {
Ops ops = Ops.create(g);
// Operations are typed classes with convenience
// builders in Ops.
Constant three = ops.constant(3);
// Single-result operations implement the Operand
// interface, so this works too.
Operand four = ops.constant(4);
// Most builders are found within a group, and accept
// Operand types as operands
Operand nine = ops.math.add(four, ops.constant(5));
// Multi-result operations however offer methods to
// select a particular result for use.
Operand result =
ops.math.add(ops.unique(s, a).y(), b);
// Optional attributes
ops.linalg.matMul(a, b, MatMul.transposeA(true));
// Naming operators
// Names can exist in a hierarchy
}
Modifier and Type | Field and Description |
---|---|
AudioOps |
audio |
BitwiseOps |
bitwise |
DataOps |
data |
DtypesOps |
dtypes |
ImageOps |
image |
IoOps |
io |
LinalgOps |
linalg |
MathOps |
math |
NnOps |
nn |
QuantizationOps |
quantization |
RandomOps |
random |
SignalOps |
signal |
SparseOps |
sparse |
StringsOps |
strings |
SummaryOps |
summary |
TrainOps |
train |
Modifier and Type | Method and Description |
---|---|
Abort |
abort(Abort.Options... options)
Builds an
Abort operation |
<T extends Number> |
all(Operand<Boolean> input,
Operand<T> axis,
All.Options... options)
Builds an
All operation |
<T extends Number> |
any(Operand<Boolean> input,
Operand<T> axis,
Any.Options... options)
Builds an
Any operation |
AssertThat |
assertThat(Operand<Boolean> condition,
Iterable<Operand<?>> data,
AssertThat.Options... options)
Builds an
AssertThat operation |
<T> Assign<T> |
assign(Operand<T> ref,
Operand<T> value,
Assign.Options... options)
Builds an
Assign operation |
<T> AssignAdd<T> |
assignAdd(Operand<T> ref,
Operand<T> value,
AssignAdd.Options... options)
Builds an
AssignAdd operation |
<T> AssignAddVariableOp |
assignAddVariableOp(Operand<?> resource,
Operand<T> value)
Builds an
AssignAddVariableOp operation |
<T> AssignSub<T> |
assignSub(Operand<T> ref,
Operand<T> value,
AssignSub.Options... options)
Builds an
AssignSub operation |
<T> AssignSubVariableOp |
assignSubVariableOp(Operand<?> resource,
Operand<T> value)
Builds an
AssignSubVariableOp operation |
<T> AssignVariableOp |
assignVariableOp(Operand<?> resource,
Operand<T> value)
Builds an
AssignVariableOp operation |
AudioOps |
audio()
Returns an API for building
audio operations |
Barrier |
barrier(List<Class<?>> componentTypes,
Barrier.Options... options)
Builds an
Barrier operation |
BarrierClose |
barrierClose(Operand<String> handle,
BarrierClose.Options... options)
Builds an
BarrierClose operation |
BarrierIncompleteSize |
barrierIncompleteSize(Operand<String> handle)
Builds an
BarrierIncompleteSize operation |
<T> BarrierInsertMany |
barrierInsertMany(Operand<String> handle,
Operand<String> keys,
Operand<T> values,
Long componentIndex)
Builds an
BarrierInsertMany operation |
BarrierReadySize |
barrierReadySize(Operand<String> handle)
Builds an
BarrierReadySize operation |
BarrierTakeMany |
barrierTakeMany(Operand<String> handle,
Operand<Integer> numElements,
List<Class<?>> componentTypes,
BarrierTakeMany.Options... options)
Builds an
BarrierTakeMany operation |
Batch |
batch(Iterable<Operand<?>> inTensors,
Long numBatchThreads,
Long maxBatchSize,
Long batchTimeoutMicros,
Long gradTimeoutMicros,
Batch.Options... options)
Builds an
Batch operation |
<T> BatchMatMulV2<T> |
batchMatMulV2(Operand<T> x,
Operand<T> y,
BatchMatMulV2.Options... options)
Builds an
BatchMatMulV2 operation |
<T,U extends Number> |
batchToSpace(Operand<T> input,
Operand<U> crops,
Long blockSize)
Builds an
BatchToSpace operation |
<T,U extends Number,V extends Number> |
batchToSpaceNd(Operand<T> input,
Operand<U> blockShape,
Operand<V> crops)
Builds an
BatchToSpaceNd operation |
<U,T> Bitcast<U> |
bitcast(Operand<T> input,
Class<U> type)
Builds an
Bitcast operation |
BitwiseOps |
bitwise()
Returns an API for building
bitwise operations |
<T extends Number> |
broadcastDynamicShape(Operand<T> s0,
Operand<T> s1)
Builds an
BroadcastDynamicShape operation |
<T,U extends Number> |
broadcastTo(Operand<T> input,
Operand<U> shape)
Builds an
BroadcastTo operation |
<T extends Number> |
bucketize(Operand<T> input,
List<Float> boundaries)
Builds an
Bucketize operation |
<T> ClipByValue<T> |
clipByValue(Operand<T> t,
Operand<T> clipValueMin,
Operand<T> clipValueMax)
Builds an
ClipByValue operation |
CombinedNonMaxSuppression |
combinedNonMaxSuppression(Operand<Float> boxes,
Operand<Float> scores,
Operand<Integer> maxOutputSizePerClass,
Operand<Integer> maxTotalSize,
Operand<Float> iouThreshold,
Operand<Float> scoreThreshold,
CombinedNonMaxSuppression.Options... options)
Builds an
CombinedNonMaxSuppression operation |
<T,U extends Number> |
concat(Iterable<Operand<T>> values,
Operand<U> axis)
Builds an
Concat operation |
Constant<Boolean> |
constant(boolean data)
Builds an
Constant operation |
Constant<Boolean> |
constant(boolean[] data)
Builds an
Constant operation |
Constant<Boolean> |
constant(boolean[][] data)
Builds an
Constant operation |
Constant<Boolean> |
constant(boolean[][][] data)
Builds an
Constant operation |
Constant<Boolean> |
constant(boolean[][][][] data)
Builds an
Constant operation |
Constant<Boolean> |
constant(boolean[][][][][] data)
Builds an
Constant operation |
Constant<Boolean> |
constant(boolean[][][][][][] data)
Builds an
Constant operation |
Constant<String> |
constant(byte[] data)
Builds an
Constant operation |
Constant<String> |
constant(byte[][] data)
Builds an
Constant operation |
Constant<String> |
constant(byte[][][] data)
Builds an
Constant operation |
Constant<String> |
constant(byte[][][][] data)
Builds an
Constant operation |
Constant<String> |
constant(byte[][][][][] data)
Builds an
Constant operation |
Constant<String> |
constant(byte[][][][][][] data)
Builds an
Constant operation |
<T> Constant<T> |
constant(Class<T> type,
long[] shape,
ByteBuffer data)
Builds an
Constant operation |
Constant<Double> |
constant(double data)
Builds an
Constant operation |
Constant<Double> |
constant(double[] data)
Builds an
Constant operation |
Constant<Double> |
constant(double[][] data)
Builds an
Constant operation |
Constant<Double> |
constant(double[][][] data)
Builds an
Constant operation |
Constant<Double> |
constant(double[][][][] data)
Builds an
Constant operation |
Constant<Double> |
constant(double[][][][][] data)
Builds an
Constant operation |
Constant<Double> |
constant(double[][][][][][] data)
Builds an
Constant operation |
Constant<Float> |
constant(float data)
Builds an
Constant operation |
Constant<Float> |
constant(float[] data)
Builds an
Constant operation |
Constant<Float> |
constant(float[][] data)
Builds an
Constant operation |
Constant<Float> |
constant(float[][][] data)
Builds an
Constant operation |
Constant<Float> |
constant(float[][][][] data)
Builds an
Constant operation |
Constant<Float> |
constant(float[][][][][] data)
Builds an
Constant operation |
Constant<Float> |
constant(float[][][][][][] data)
Builds an
Constant operation |
Constant<Integer> |
constant(int data)
Builds an
Constant operation |
Constant<Integer> |
constant(int[] data)
Builds an
Constant operation |
Constant<Integer> |
constant(int[][] data)
Builds an
Constant operation |
Constant<Integer> |
constant(int[][][] data)
Builds an
Constant operation |
Constant<Integer> |
constant(int[][][][] data)
Builds an
Constant operation |
Constant<Integer> |
constant(int[][][][][] data)
Builds an
Constant operation |
Constant<Integer> |
constant(int[][][][][][] data)
Builds an
Constant operation |
Constant<Long> |
constant(long data)
Builds an
Constant operation |
Constant<Long> |
constant(long[] data)
Builds an
Constant operation |
Constant<Long> |
constant(long[][] data)
Builds an
Constant operation |
Constant<Long> |
constant(long[][][] data)
Builds an
Constant operation |
Constant<Long> |
constant(long[][][][] data)
Builds an
Constant operation |
Constant<Long> |
constant(long[][][][][] data)
Builds an
Constant operation |
Constant<Long> |
constant(long[][][][][][] data)
Builds an
Constant operation |
Constant<Double> |
constant(long[] shape,
DoubleBuffer data)
Builds an
Constant operation |
Constant<Float> |
constant(long[] shape,
FloatBuffer data)
Builds an
Constant operation |
Constant<Integer> |
constant(long[] shape,
IntBuffer data)
Builds an
Constant operation |
Constant<Long> |
constant(long[] shape,
LongBuffer data)
Builds an
Constant operation |
<T> Constant<T> |
constant(Object object,
Class<T> type)
Builds an
Constant operation |
Constant<String> |
constant(String data)
Builds an
Constant operation |
Constant<String> |
constant(String data,
Charset charset)
Builds an
Constant operation |
ConsumeMutexLock |
consumeMutexLock(Operand<?> mutexLock)
Builds an
ConsumeMutexLock operation |
ControlTrigger |
controlTrigger()
Builds an
ControlTrigger operation |
<T extends Number> |
countUpTo(Operand<T> ref,
Long limit)
Builds an
CountUpTo operation |
static Ops |
create()
Creates an API for building operations in the default eager execution environment
|
static Ops |
create(ExecutionEnvironment env)
Creates an API for building operations in the provided execution environment
|
<T extends Number> |
cudnnRNNCanonicalToParamsV2(Operand<Integer> numLayers,
Operand<Integer> numUnits,
Operand<Integer> inputSize,
Iterable<Operand<T>> weights,
Iterable<Operand<T>> biases,
CudnnRNNCanonicalToParamsV2.Options... options)
Builds an
CudnnRNNCanonicalToParamsV2 operation |
<T extends Number> |
cudnnRNNParamsToCanonicalV2(Operand<Integer> numLayers,
Operand<Integer> numUnits,
Operand<Integer> inputSize,
Operand<T> params,
Long numParamsWeights,
Long numParamsBiases,
CudnnRNNParamsToCanonicalV2.Options... options)
Builds an
CudnnRNNParamsToCanonicalV2 operation |
DataOps |
data()
Returns an API for building
data operations |
<T extends Number> |
decodePaddedRaw(Operand<String> inputBytes,
Operand<Integer> fixedLength,
Class<T> outType,
DecodePaddedRaw.Options... options)
Builds an
DecodePaddedRaw operation |
<T> DeepCopy<T> |
deepCopy(Operand<T> x)
Builds an
DeepCopy operation |
DeleteSessionTensor |
deleteSessionTensor(Operand<String> handle)
Builds an
DeleteSessionTensor operation |
DestroyResourceOp |
destroyResourceOp(Operand<?> resource,
DestroyResourceOp.Options... options)
Builds an
DestroyResourceOp operation |
<T> DestroyTemporaryVariable<T> |
destroyTemporaryVariable(Operand<T> ref,
String varName)
Builds an
DestroyTemporaryVariable operation |
<T extends Number> |
drawBoundingBoxesV2(Operand<T> images,
Operand<Float> boxes,
Operand<Float> colors)
Builds an
DrawBoundingBoxesV2 operation |
DtypesOps |
dtypes()
Returns an API for building
dtypes operations |
<T> DynamicPartition<T> |
dynamicPartition(Operand<T> data,
Operand<Integer> partitions,
Long numPartitions)
Builds an
DynamicPartition operation |
<T> DynamicStitch<T> |
dynamicStitch(Iterable<Operand<Integer>> indices,
Iterable<Operand<T>> data)
Builds an
DynamicStitch operation |
<T> EditDistance |
editDistance(Operand<Long> hypothesisIndices,
Operand<T> hypothesisValues,
Operand<Long> hypothesisShape,
Operand<Long> truthIndices,
Operand<T> truthValues,
Operand<Long> truthShape,
EditDistance.Options... options)
Builds an
EditDistance operation |
<T> Einsum<T> |
einsum(Iterable<Operand<T>> inputs,
String equation)
Builds an
Einsum operation |
<T> Empty<T> |
empty(Operand<Integer> shape,
Class<T> dtype,
Empty.Options... options)
Builds an
Empty operation |
<T extends Number,U> |
emptyTensorList(Operand<T> elementShape,
Operand<Integer> maxNumElements,
Class<U> elementDtype)
Builds an
EmptyTensorList operation |
<T> EnsureShape<T> |
ensureShape(Operand<T> input,
Shape shape)
Builds an
EnsureShape operation |
<T,U extends Number> |
euclideanNorm(Operand<T> input,
Operand<U> axis,
EuclideanNorm.Options... options)
Builds an
EuclideanNorm operation |
<T,U extends Number> |
expandDims(Operand<T> input,
Operand<U> axis)
Builds an
ExpandDims operation |
<T extends Number> |
extractVolumePatches(Operand<T> input,
List<Long> ksizes,
List<Long> strides,
String padding)
Builds an
ExtractVolumePatches operation |
<U,T extends Number> |
fill(Operand<T> dims,
Operand<U> value)
Builds an
Fill operation |
<T> Fingerprint |
fingerprint(Operand<T> data,
Operand<String> method)
Builds an
Fingerprint operation |
<T extends Number,U extends Number> |
fusedBatchNormGradV3(Operand<T> yBackprop,
Operand<T> x,
Operand<Float> scale,
Operand<U> reserveSpace1,
Operand<U> reserveSpace2,
Operand<U> reserveSpace3,
FusedBatchNormGradV3.Options... options)
Builds an
FusedBatchNormGradV3 operation |
<T extends Number,U extends Number> |
fusedBatchNormV3(Operand<T> x,
Operand<U> scale,
Operand<U> offset,
Operand<U> mean,
Operand<U> variance,
FusedBatchNormV3.Options... options)
Builds an
FusedBatchNormV3 operation |
<T,U extends Number,V extends Number> |
gather(Operand<T> params,
Operand<U> indices,
Operand<V> axis,
Gather.Options... options)
Builds an
Gather operation |
<T,U extends Number> |
gatherNd(Operand<T> params,
Operand<U> indices)
Builds an
GatherNd operation |
<T> GetSessionHandle |
getSessionHandle(Operand<T> value)
Builds an
GetSessionHandle operation |
<T> GetSessionTensor<T> |
getSessionTensor(Operand<String> handle,
Class<T> dtype)
Builds an
GetSessionTensor operation |
Gradients |
gradients(Iterable<? extends Operand<?>> y,
Iterable<? extends Operand<?>> x,
Gradients.Options... options)
Builds an
Gradients operation |
Gradients |
gradients(Operand<?> y,
Iterable<? extends Operand<?>> x,
Gradients.Options... options)
Builds an
Gradients operation |
<T> GuaranteeConst<T> |
guaranteeConst(Operand<T> input)
Builds an
GuaranteeConst operation |
<T,U> HashTable |
hashTable(Class<T> keyDtype,
Class<U> valueDtype,
HashTable.Options... options)
Builds an
HashTable operation |
<T extends Number> |
histogramFixedWidth(Operand<T> values,
Operand<T> valueRange,
Operand<Integer> nbins)
Builds an
HistogramFixedWidth operation |
<U extends Number,T extends Number> |
histogramFixedWidth(Operand<T> values,
Operand<T> valueRange,
Operand<Integer> nbins,
Class<U> dtype)
Builds an
HistogramFixedWidth operation |
<T> Identity<T> |
identity(Operand<T> input)
Builds an
Identity operation |
IdentityN |
identityN(Iterable<Operand<?>> input)
Builds an
IdentityN operation |
ImageOps |
image()
Returns an API for building
image operations |
<T> ImmutableConst<T> |
immutableConst(Class<T> dtype,
Shape shape,
String memoryRegionName)
Builds an
ImmutableConst operation |
<T,U> InitializeTable |
initializeTable(Operand<?> tableHandle,
Operand<T> keys,
Operand<U> values)
Builds an
InitializeTable operation |
InitializeTableFromTextFile |
initializeTableFromTextFile(Operand<?> tableHandle,
Operand<String> filename,
Long keyIndex,
Long valueIndex,
InitializeTableFromTextFile.Options... options)
Builds an
InitializeTableFromTextFile operation |
<T> InplaceAdd<T> |
inplaceAdd(Operand<T> x,
Operand<Integer> i,
Operand<T> v)
Builds an
InplaceAdd operation |
<T> InplaceSub<T> |
inplaceSub(Operand<T> x,
Operand<Integer> i,
Operand<T> v)
Builds an
InplaceSub operation |
<T> InplaceUpdate<T> |
inplaceUpdate(Operand<T> x,
Operand<Integer> i,
Operand<T> v)
Builds an
InplaceUpdate operation |
IoOps |
io()
Returns an API for building
io operations |
<T> IsVariableInitialized |
isVariableInitialized(Operand<T> ref)
Builds an
IsVariableInitialized operation |
LinalgOps |
linalg()
Returns an API for building
linalg operations |
<T extends Number,U extends Number> |
linSpace(Operand<T> start,
Operand<T> stop,
Operand<U> num)
Builds an
LinSpace operation |
<T,U> LookupTableExport<T,U> |
lookupTableExport(Operand<?> tableHandle,
Class<T> Tkeys,
Class<U> Tvalues)
Builds an
LookupTableExport operation |
<U,T> LookupTableFind<U> |
lookupTableFind(Operand<?> tableHandle,
Operand<T> keys,
Operand<U> defaultValue)
Builds an
LookupTableFind operation |
<T,U> LookupTableImport |
lookupTableImport(Operand<?> tableHandle,
Operand<T> keys,
Operand<U> values)
Builds an
LookupTableImport operation |
<T,U> LookupTableInsert |
lookupTableInsert(Operand<?> tableHandle,
Operand<T> keys,
Operand<U> values)
Builds an
LookupTableInsert operation |
LookupTableSize |
lookupTableSize(Operand<?> tableHandle)
Builds an
LookupTableSize operation |
LoopCond |
loopCond(Operand<Boolean> input)
Builds an
LoopCond operation |
<T> Lu<T,Integer> |
lu(Operand<T> input)
Builds an
Lu operation |
<T,U extends Number> |
lu(Operand<T> input,
Class<U> outputIdxType)
Builds an
Lu operation |
MapClear |
mapClear(List<Class<?>> dtypes,
MapClear.Options... options)
Builds an
MapClear operation |
MapIncompleteSize |
mapIncompleteSize(List<Class<?>> dtypes,
MapIncompleteSize.Options... options)
Builds an
MapIncompleteSize operation |
MapPeek |
mapPeek(Operand<Long> key,
Operand<Integer> indices,
List<Class<?>> dtypes,
MapPeek.Options... options)
Builds an
MapPeek operation |
MapSize |
mapSize(List<Class<?>> dtypes,
MapSize.Options... options)
Builds an
MapSize operation |
MapStage |
mapStage(Operand<Long> key,
Operand<Integer> indices,
Iterable<Operand<?>> values,
List<Class<?>> dtypes,
MapStage.Options... options)
Builds an
MapStage operation |
MapUnstage |
mapUnstage(Operand<Long> key,
Operand<Integer> indices,
List<Class<?>> dtypes,
MapUnstage.Options... options)
Builds an
MapUnstage operation |
MapUnstageNoKey |
mapUnstageNoKey(Operand<Integer> indices,
List<Class<?>> dtypes,
MapUnstageNoKey.Options... options)
Builds an
MapUnstageNoKey operation |
MathOps |
math()
Returns an API for building
math operations |
<T> MatrixDiagPartV2<T> |
matrixDiagPartV2(Operand<T> input,
Operand<Integer> k,
Operand<T> paddingValue)
Builds an
MatrixDiagPartV2 operation |
<T> MatrixDiagV2<T> |
matrixDiagV2(Operand<T> diagonal,
Operand<Integer> k,
Operand<Integer> numRows,
Operand<Integer> numCols,
Operand<T> paddingValue)
Builds an
MatrixDiagV2 operation |
<T> MatrixSetDiagV2<T> |
matrixSetDiagV2(Operand<T> input,
Operand<T> diagonal,
Operand<Integer> k)
Builds an
MatrixSetDiagV2 operation |
<T,U extends Number> |
max(Operand<T> input,
Operand<U> axis,
Max.Options... options)
Builds an
Max operation |
<T> Merge<T> |
merge(Iterable<Operand<T>> inputs)
Builds an
Merge operation |
<T,U extends Number> |
min(Operand<T> input,
Operand<U> axis,
Min.Options... options)
Builds an
Min operation |
<T,U extends Number> |
mirrorPad(Operand<T> input,
Operand<U> paddings,
String mode)
Builds an
MirrorPad operation |
<T> MulNoNan<T> |
mulNoNan(Operand<T> x,
Operand<T> y)
Builds an
MulNoNan operation |
<T,U> MutableDenseHashTable |
mutableDenseHashTable(Operand<T> emptyKey,
Operand<T> deletedKey,
Class<U> valueDtype,
MutableDenseHashTable.Options... options)
Builds an
MutableDenseHashTable operation |
<T,U> MutableHashTable |
mutableHashTable(Class<T> keyDtype,
Class<U> valueDtype,
MutableHashTable.Options... options)
Builds an
MutableHashTable operation |
<T,U> MutableHashTableOfTensors |
mutableHashTableOfTensors(Class<T> keyDtype,
Class<U> valueDtype,
MutableHashTableOfTensors.Options... options)
Builds an
MutableHashTableOfTensors operation |
Mutex |
mutex(Mutex.Options... options)
Builds an
Mutex operation |
MutexLock |
mutexLock(Operand<?> mutex)
Builds an
MutexLock operation |
<T extends Number> |
nextAfter(Operand<T> x1,
Operand<T> x2)
Builds an
NextAfter operation |
<T> NextIteration<T> |
nextIteration(Operand<T> data)
Builds an
NextIteration operation |
NnOps |
nn()
Returns an API for building
nn operations |
<T extends Number> |
nonMaxSuppressionV5(Operand<T> boxes,
Operand<T> scores,
Operand<Integer> maxOutputSize,
Operand<T> iouThreshold,
Operand<T> scoreThreshold,
Operand<T> softNmsSigma,
NonMaxSuppressionV5.Options... options)
Builds an
NonMaxSuppressionV5 operation |
NoOp |
noOp()
Builds an
NoOp operation |
<U,T extends Number> |
oneHot(Operand<T> indices,
Operand<Integer> depth,
Operand<U> onValue,
Operand<U> offValue,
OneHot.Options... options)
Builds an
OneHot operation |
<T> OnesLike<T> |
onesLike(Operand<T> x)
Builds an
OnesLike operation |
OrderedMapClear |
orderedMapClear(List<Class<?>> dtypes,
OrderedMapClear.Options... options)
Builds an
OrderedMapClear operation |
OrderedMapIncompleteSize |
orderedMapIncompleteSize(List<Class<?>> dtypes,
OrderedMapIncompleteSize.Options... options)
Builds an
OrderedMapIncompleteSize operation |
OrderedMapPeek |
orderedMapPeek(Operand<Long> key,
Operand<Integer> indices,
List<Class<?>> dtypes,
OrderedMapPeek.Options... options)
Builds an
OrderedMapPeek operation |
OrderedMapSize |
orderedMapSize(List<Class<?>> dtypes,
OrderedMapSize.Options... options)
Builds an
OrderedMapSize operation |
OrderedMapStage |
orderedMapStage(Operand<Long> key,
Operand<Integer> indices,
Iterable<Operand<?>> values,
List<Class<?>> dtypes,
OrderedMapStage.Options... options)
Builds an
OrderedMapStage operation |
OrderedMapUnstage |
orderedMapUnstage(Operand<Long> key,
Operand<Integer> indices,
List<Class<?>> dtypes,
OrderedMapUnstage.Options... options)
Builds an
OrderedMapUnstage operation |
OrderedMapUnstageNoKey |
orderedMapUnstageNoKey(Operand<Integer> indices,
List<Class<?>> dtypes,
OrderedMapUnstageNoKey.Options... options)
Builds an
OrderedMapUnstageNoKey operation |
<T,U extends Number> |
pad(Operand<T> input,
Operand<U> paddings,
Operand<T> constantValues)
Builds an
Pad operation |
<T> ParallelConcat<T> |
parallelConcat(Iterable<Operand<T>> values,
Shape shape)
Builds an
ParallelConcat operation |
<T> ParallelDynamicStitch<T> |
parallelDynamicStitch(Iterable<Operand<Integer>> indices,
Iterable<Operand<T>> data)
Builds an
ParallelDynamicStitch operation |
<T> Placeholder<T> |
placeholder(Class<T> dtype,
Placeholder.Options... options)
Builds an
Placeholder operation |
<T> PlaceholderWithDefault<T> |
placeholderWithDefault(Operand<T> input,
Shape shape)
Builds an
PlaceholderWithDefault operation |
Print |
print(Operand<String> input,
Print.Options... options)
Builds an
Print operation |
<T,U extends Number> |
prod(Operand<T> input,
Operand<U> axis,
Prod.Options... options)
Builds an
Prod operation |
QuantizationOps |
quantization()
Returns an API for building
quantization operations |
<T> QuantizedConcat<T> |
quantizedConcat(Operand<Integer> concatDim,
Iterable<Operand<T>> values,
Iterable<Operand<Float>> inputMins,
Iterable<Operand<Float>> inputMaxes)
Builds an
QuantizedConcat operation |
<T,U extends Number> |
quantizedConcatV2(Iterable<Operand<T>> values,
Operand<U> axis,
Iterable<Operand<Float>> inputMins,
Iterable<Operand<Float>> inputMaxes)
Builds an
QuantizedConcatV2 operation |
<T,U extends Number> |
quantizedReshape(Operand<T> tensor,
Operand<U> shape,
Operand<Float> inputMin,
Operand<Float> inputMax)
Builds an
QuantizedReshape operation |
RandomOps |
random()
Returns an API for building
random operations |
<T extends Number> |
range(Operand<T> start,
Operand<T> limit,
Operand<T> delta)
Builds an
Range operation |
<T> Rank |
rank(Operand<T> input)
Builds an
Rank operation |
<T> ReadVariableOp<T> |
readVariableOp(Operand<?> resource,
Class<T> dtype)
Builds an
ReadVariableOp operation |
<T extends Number> |
reduceAll(Operand<Boolean> input,
Operand<T> axis,
ReduceAll.Options... options)
Builds an
ReduceAll operation |
<T extends Number> |
reduceAny(Operand<Boolean> input,
Operand<T> axis,
ReduceAny.Options... options)
Builds an
ReduceAny operation |
<T,U extends Number> |
reduceMax(Operand<T> input,
Operand<U> axis,
ReduceMax.Options... options)
Builds an
ReduceMax operation |
<T,U extends Number> |
reduceMin(Operand<T> input,
Operand<U> axis,
ReduceMin.Options... options)
Builds an
ReduceMin operation |
<T,U extends Number> |
reduceProd(Operand<T> input,
Operand<U> axis,
ReduceProd.Options... options)
Builds an
ReduceProd operation |
<T,U extends Number> |
reduceSum(Operand<T> input,
Operand<U> axis,
ReduceSum.Options... options)
Builds an
ReduceSum operation |
<T> RefNextIteration<T> |
refNextIteration(Operand<T> data)
Builds an
RefNextIteration operation |
<T> RefSelect<T> |
refSelect(Operand<Integer> index,
Iterable<Operand<T>> inputs)
Builds an
RefSelect operation |
<T> RefSwitch<T> |
refSwitch(Operand<T> data,
Operand<Boolean> pred)
Builds an
RefSwitch operation |
RemoteFusedGraphExecute |
remoteFusedGraphExecute(Iterable<Operand<?>> inputs,
List<Class<?>> Toutputs,
String serializedRemoteFusedGraphExecuteInfo)
Builds an
RemoteFusedGraphExecute operation |
<T,U extends Number> |
reshape(Operand<T> tensor,
Operand<U> shape)
Builds an
Reshape operation |
<T> ResourceApplyAdamWithAmsgrad |
resourceApplyAdamWithAmsgrad(Operand<?> var,
Operand<?> m,
Operand<?> v,
Operand<?> vhat,
Operand<T> beta1Power,
Operand<T> beta2Power,
Operand<T> lr,
Operand<T> beta1,
Operand<T> beta2,
Operand<T> epsilon,
Operand<T> grad,
ResourceApplyAdamWithAmsgrad.Options... options)
Builds an
ResourceApplyAdamWithAmsgrad operation |
<T> ResourceApplyKerasMomentum |
resourceApplyKerasMomentum(Operand<?> var,
Operand<?> accum,
Operand<T> lr,
Operand<T> grad,
Operand<T> momentum,
ResourceApplyKerasMomentum.Options... options)
Builds an
ResourceApplyKerasMomentum operation |
<T extends Number> |
resourceCountUpTo(Operand<?> resource,
Long limit,
Class<T> T)
Builds an
ResourceCountUpTo operation |
<U,T extends Number> |
resourceGather(Operand<?> resource,
Operand<T> indices,
Class<U> dtype,
ResourceGather.Options... options)
Builds an
ResourceGather operation |
<U,T extends Number> |
resourceGatherNd(Operand<?> resource,
Operand<T> indices,
Class<U> dtype)
Builds an
ResourceGatherNd operation |
<T extends Number,U> |
resourceScatterAdd(Operand<?> resource,
Operand<T> indices,
Operand<U> updates)
Builds an
ResourceScatterAdd operation |
<T extends Number,U> |
resourceScatterDiv(Operand<?> resource,
Operand<T> indices,
Operand<U> updates)
Builds an
ResourceScatterDiv operation |
<T extends Number,U> |
resourceScatterMax(Operand<?> resource,
Operand<T> indices,
Operand<U> updates)
Builds an
ResourceScatterMax operation |
<T extends Number,U> |
resourceScatterMin(Operand<?> resource,
Operand<T> indices,
Operand<U> updates)
Builds an
ResourceScatterMin operation |
<T extends Number,U> |
resourceScatterMul(Operand<?> resource,
Operand<T> indices,
Operand<U> updates)
Builds an
ResourceScatterMul operation |
<T extends Number,U> |
resourceScatterNdAdd(Operand<?> ref,
Operand<T> indices,
Operand<U> updates,
ResourceScatterNdAdd.Options... options)
Builds an
ResourceScatterNdAdd operation |
<T extends Number,U> |
resourceScatterNdSub(Operand<?> ref,
Operand<T> indices,
Operand<U> updates,
ResourceScatterNdSub.Options... options)
Builds an
ResourceScatterNdSub operation |
<T extends Number,U> |
resourceScatterNdUpdate(Operand<?> ref,
Operand<T> indices,
Operand<U> updates,
ResourceScatterNdUpdate.Options... options)
Builds an
ResourceScatterNdUpdate operation |
<T extends Number,U> |
resourceScatterSub(Operand<?> resource,
Operand<T> indices,
Operand<U> updates)
Builds an
ResourceScatterSub operation |
<T extends Number,U> |
resourceScatterUpdate(Operand<?> resource,
Operand<T> indices,
Operand<U> updates)
Builds an
ResourceScatterUpdate operation |
<T,U extends Number> |
resourceSparseApplyKerasMomentum(Operand<?> var,
Operand<?> accum,
Operand<T> lr,
Operand<T> grad,
Operand<U> indices,
Operand<T> momentum,
ResourceSparseApplyKerasMomentum.Options... options)
Builds an
ResourceSparseApplyKerasMomentum operation |
<T extends Number,U> |
resourceStridedSliceAssign(Operand<?> ref,
Operand<T> begin,
Operand<T> end,
Operand<T> strides,
Operand<U> value,
ResourceStridedSliceAssign.Options... options)
Builds an
ResourceStridedSliceAssign operation |
<T,U extends Number> |
reverse(Operand<T> tensor,
Operand<U> axis)
Builds an
Reverse operation |
<T,U extends Number> |
reverseSequence(Operand<T> input,
Operand<U> seqLengths,
Long seqDim,
ReverseSequence.Options... options)
Builds an
ReverseSequence operation |
<T,U extends Number,V extends Number> |
roll(Operand<T> input,
Operand<U> shift,
Operand<V> axis)
Builds an
Roll operation |
Rpc |
rpc(Operand<String> address,
Operand<String> method,
Operand<String> request,
Rpc.Options... options)
Builds an
Rpc operation |
<T extends Number> |
scaleAndTranslate(Operand<T> images,
Operand<Integer> size,
Operand<Float> scale,
Operand<Float> translation,
ScaleAndTranslate.Options... options)
Builds an
ScaleAndTranslate operation |
<T,U extends Number> |
scatterAdd(Operand<T> ref,
Operand<U> indices,
Operand<T> updates,
ScatterAdd.Options... options)
Builds an
ScatterAdd operation |
<T,U extends Number> |
scatterDiv(Operand<T> ref,
Operand<U> indices,
Operand<T> updates,
ScatterDiv.Options... options)
Builds an
ScatterDiv operation |
<T extends Number,U extends Number> |
scatterMax(Operand<T> ref,
Operand<U> indices,
Operand<T> updates,
ScatterMax.Options... options)
Builds an
ScatterMax operation |
<T extends Number,U extends Number> |
scatterMin(Operand<T> ref,
Operand<U> indices,
Operand<T> updates,
ScatterMin.Options... options)
Builds an
ScatterMin operation |
<T,U extends Number> |
scatterMul(Operand<T> ref,
Operand<U> indices,
Operand<T> updates,
ScatterMul.Options... options)
Builds an
ScatterMul operation |
<U,T extends Number> |
scatterNd(Operand<T> indices,
Operand<U> updates,
Operand<T> shape)
Builds an
ScatterNd operation |
<T,U extends Number> |
scatterNdAdd(Operand<T> ref,
Operand<U> indices,
Operand<T> updates,
ScatterNdAdd.Options... options)
Builds an
ScatterNdAdd operation |
<T,U extends Number> |
scatterNdNonAliasingAdd(Operand<T> input,
Operand<U> indices,
Operand<T> updates)
Builds an
ScatterNdNonAliasingAdd operation |
<T,U extends Number> |
scatterNdSub(Operand<T> ref,
Operand<U> indices,
Operand<T> updates,
ScatterNdSub.Options... options)
Builds an
ScatterNdSub operation |
<T,U extends Number> |
scatterNdUpdate(Operand<T> ref,
Operand<U> indices,
Operand<T> updates,
ScatterNdUpdate.Options... options)
Builds an
ScatterNdUpdate operation |
<T,U extends Number> |
scatterSub(Operand<T> ref,
Operand<U> indices,
Operand<T> updates,
ScatterSub.Options... options)
Builds an
ScatterSub operation |
<T,U extends Number> |
scatterUpdate(Operand<T> ref,
Operand<U> indices,
Operand<T> updates,
ScatterUpdate.Options... options)
Builds an
ScatterUpdate operation |
Scope |
scope()
Returns the current
scope of this API |
<T> SelectV2<T> |
selectV2(Operand<Boolean> condition,
Operand<T> t,
Operand<T> e)
Builds an
SelectV2 operation |
<T> SetDiff1d<T,Integer> |
setDiff1d(Operand<T> x,
Operand<T> y)
Builds an
SetDiff1d operation |
<T,U extends Number> |
setDiff1d(Operand<T> x,
Operand<T> y,
Class<U> outIdx)
Builds an
SetDiff1d operation |
<T> SetSize |
setSize(Operand<Long> setIndices,
Operand<T> setValues,
Operand<Long> setShape,
SetSize.Options... options)
Builds an
SetSize operation |
<T> Shape<Integer> |
shape(Operand<T> input)
Builds an
Shape operation |
<U extends Number,T> |
shape(Operand<T> input,
Class<U> outType)
Builds an
Shape operation |
<T> ShapeN<Integer> |
shapeN(Iterable<Operand<T>> input)
Builds an
ShapeN operation |
<U extends Number,T> |
shapeN(Iterable<Operand<T>> input,
Class<U> outType)
Builds an
ShapeN operation |
SignalOps |
signal()
Returns an API for building
signal operations |
<T> Size<Integer> |
size(Operand<T> input)
Builds an
Size operation |
<U extends Number,T> |
size(Operand<T> input,
Class<U> outType)
Builds an
Size operation |
Skipgram |
skipgram(String filename,
Long batchSize,
Skipgram.Options... options)
Builds an
Skipgram operation |
<T,U extends Number> |
slice(Operand<T> input,
Operand<U> begin,
Operand<U> size)
Builds an
Slice operation |
<T> Snapshot<T> |
snapshot(Operand<T> input)
Builds an
Snapshot operation |
<T,U extends Number,V extends Number> |
spaceToBatchNd(Operand<T> input,
Operand<U> blockShape,
Operand<V> paddings)
Builds an
SpaceToBatchNd operation |
SparseOps |
sparse()
Returns an API for building
sparse operations |
<T> Split<T> |
split(Operand<Integer> axis,
Operand<T> value,
Long numSplit)
Builds an
Split operation |
<T,U extends Number> |
splitV(Operand<T> value,
Operand<U> sizeSplits,
Operand<Integer> axis,
Long numSplit)
Builds an
SplitV operation |
<T> Squeeze<T> |
squeeze(Operand<T> input,
Squeeze.Options... options)
Builds an
Squeeze operation |
<T> Stack<T> |
stack(Iterable<Operand<T>> values,
Stack.Options... options)
Builds an
Stack operation |
Stage |
stage(Iterable<Operand<?>> values,
Stage.Options... options)
Builds an
Stage operation |
StageClear |
stageClear(List<Class<?>> dtypes,
StageClear.Options... options)
Builds an
StageClear operation |
StagePeek |
stagePeek(Operand<Integer> index,
List<Class<?>> dtypes,
StagePeek.Options... options)
Builds an
StagePeek operation |
StageSize |
stageSize(List<Class<?>> dtypes,
StageSize.Options... options)
Builds an
StageSize operation |
<T extends Number,U extends Number> |
statefulRandomBinomial(Operand<?> resource,
Operand<Long> algorithm,
Operand<T> shape,
Operand<U> counts,
Operand<U> probs)
Builds an
StatefulRandomBinomial operation |
<V extends Number,T extends Number,U extends Number> |
statefulRandomBinomial(Operand<?> resource,
Operand<Long> algorithm,
Operand<T> shape,
Operand<U> counts,
Operand<U> probs,
Class<V> dtype)
Builds an
StatefulRandomBinomial operation |
<T> StatefulStandardNormal<Float> |
statefulStandardNormal(Operand<?> resource,
Operand<T> shape)
Builds an
StatefulStandardNormal operation |
<U,T> StatefulStandardNormal<U> |
statefulStandardNormal(Operand<?> resource,
Operand<T> shape,
Class<U> dtype)
Builds an
StatefulStandardNormal operation |
<T> StatefulStandardNormalV2<Float> |
statefulStandardNormalV2(Operand<?> resource,
Operand<Long> algorithm,
Operand<T> shape)
Builds an
StatefulStandardNormalV2 operation |
<U,T> StatefulStandardNormalV2<U> |
statefulStandardNormalV2(Operand<?> resource,
Operand<Long> algorithm,
Operand<T> shape,
Class<U> dtype)
Builds an
StatefulStandardNormalV2 operation |
<T> StopGradient<T> |
stopGradient(Operand<T> input)
Builds an
StopGradient operation |
<T,U extends Number> |
stridedSlice(Operand<T> input,
Operand<U> begin,
Operand<U> end,
Operand<U> strides,
StridedSlice.Options... options)
Builds an
StridedSlice operation |
<T,U extends Number> |
stridedSliceAssign(Operand<T> ref,
Operand<U> begin,
Operand<U> end,
Operand<U> strides,
Operand<T> value,
StridedSliceAssign.Options... options)
Builds an
StridedSliceAssign operation |
<U,T extends Number> |
stridedSliceGrad(Operand<T> shape,
Operand<T> begin,
Operand<T> end,
Operand<T> strides,
Operand<U> dy,
StridedSliceGrad.Options... options)
Builds an
StridedSliceGrad operation |
StringLower |
stringLower(Operand<String> input,
StringLower.Options... options)
Builds an
StringLower operation |
<T extends Number> |
stringNGrams(Operand<String> data,
Operand<T> dataSplits,
String separator,
List<Long> ngramWidths,
String leftPad,
String rightPad,
Long padWidth,
Boolean preserveShortSequences)
Builds an
StringNGrams operation |
StringsOps |
strings()
Returns an API for building
strings operations |
StringUpper |
stringUpper(Operand<String> input,
StringUpper.Options... options)
Builds an
StringUpper operation |
<T,U extends Number> |
sum(Operand<T> input,
Operand<U> axis,
Sum.Options... options)
Builds an
Sum operation |
SummaryOps |
summary()
Returns an API for building
summary operations |
<T> SwitchCond<T> |
switchCond(Operand<T> data,
Operand<Boolean> pred)
Builds an
SwitchCond operation |
<T> TemporaryVariable<T> |
temporaryVariable(Shape shape,
Class<T> dtype,
TemporaryVariable.Options... options)
Builds an
TemporaryVariable operation |
<T> TensorArray |
tensorArray(Operand<Integer> size,
Class<T> dtype,
TensorArray.Options... options)
Builds an
TensorArray operation |
TensorArrayClose |
tensorArrayClose(Operand<?> handle)
Builds an
TensorArrayClose operation |
<T> TensorArrayConcat<T> |
tensorArrayConcat(Operand<?> handle,
Operand<Float> flowIn,
Class<T> dtype,
TensorArrayConcat.Options... options)
Builds an
TensorArrayConcat operation |
<T> TensorArrayGather<T> |
tensorArrayGather(Operand<?> handle,
Operand<Integer> indices,
Operand<Float> flowIn,
Class<T> dtype,
TensorArrayGather.Options... options)
Builds an
TensorArrayGather operation |
TensorArrayGrad |
tensorArrayGrad(Operand<?> handle,
Operand<Float> flowIn,
String source)
Builds an
TensorArrayGrad operation |
TensorArrayGradWithShape |
tensorArrayGradWithShape(Operand<?> handle,
Operand<Float> flowIn,
Operand<Integer> shapeToPrepend,
String source)
Builds an
TensorArrayGradWithShape operation |
<T> TensorArrayPack<T> |
tensorArrayPack(Operand<String> handle,
Operand<Float> flowIn,
Class<T> dtype,
TensorArrayPack.Options... options)
Builds an
TensorArrayPack operation |
<T> TensorArrayRead<T> |
tensorArrayRead(Operand<?> handle,
Operand<Integer> index,
Operand<Float> flowIn,
Class<T> dtype)
Builds an
TensorArrayRead operation |
<T> TensorArrayScatter |
tensorArrayScatter(Operand<?> handle,
Operand<Integer> indices,
Operand<T> value,
Operand<Float> flowIn)
Builds an
TensorArrayScatter operation |
TensorArraySize |
tensorArraySize(Operand<?> handle,
Operand<Float> flowIn)
Builds an
TensorArraySize operation |
<T> TensorArraySplit |
tensorArraySplit(Operand<?> handle,
Operand<T> value,
Operand<Long> lengths,
Operand<Float> flowIn)
Builds an
TensorArraySplit operation |
<T> TensorArrayUnpack |
tensorArrayUnpack(Operand<String> handle,
Operand<T> value,
Operand<Float> flowIn)
Builds an
TensorArrayUnpack operation |
<T> TensorArrayWrite |
tensorArrayWrite(Operand<?> handle,
Operand<Integer> index,
Operand<T> value,
Operand<Float> flowIn)
Builds an
TensorArrayWrite operation |
<T> TensorListConcat<T> |
tensorListConcat(Operand<?> inputHandle,
Class<T> elementDtype,
TensorListConcat.Options... options)
Builds an
TensorListConcat operation |
<T> TensorListConcatLists |
tensorListConcatLists(Operand<?> inputA,
Operand<?> inputB,
Class<T> elementDtype)
Builds an
TensorListConcatLists operation |
<U,T extends Number> |
tensorListConcatV2(Operand<?> inputHandle,
Operand<T> elementShape,
Operand<Long> leadingDims,
Class<U> elementDtype)
Builds an
TensorListConcatV2 operation |
<T extends Number> |
tensorListElementShape(Operand<?> inputHandle,
Class<T> shapeType)
Builds an
TensorListElementShape operation |
<T,U extends Number> |
tensorListFromTensor(Operand<T> tensor,
Operand<U> elementShape)
Builds an
TensorListFromTensor operation |
<T> TensorListGather<T> |
tensorListGather(Operand<?> inputHandle,
Operand<Integer> indices,
Operand<Integer> elementShape,
Class<T> elementDtype)
Builds an
TensorListGather operation |
<T> TensorListGetItem<T> |
tensorListGetItem(Operand<?> inputHandle,
Operand<Integer> index,
Operand<Integer> elementShape,
Class<T> elementDtype)
Builds an
TensorListGetItem operation |
TensorListLength |
tensorListLength(Operand<?> inputHandle)
Builds an
TensorListLength operation |
<T> TensorListPopBack<T> |
tensorListPopBack(Operand<?> inputHandle,
Operand<Integer> elementShape,
Class<T> elementDtype)
Builds an
TensorListPopBack operation |
<T> TensorListPushBack |
tensorListPushBack(Operand<?> inputHandle,
Operand<T> tensor)
Builds an
TensorListPushBack operation |
<T> TensorListPushBackBatch |
tensorListPushBackBatch(Operand<?> inputHandles,
Operand<T> tensor)
Builds an
TensorListPushBackBatch operation |
<T extends Number,U> |
tensorListReserve(Operand<T> elementShape,
Operand<Integer> numElements,
Class<U> elementDtype)
Builds an
TensorListReserve operation |
TensorListResize |
tensorListResize(Operand<?> inputHandle,
Operand<Integer> size)
Builds an
TensorListResize operation |
<T,U extends Number> |
tensorListScatter(Operand<T> tensor,
Operand<Integer> indices,
Operand<U> elementShape)
Builds an
TensorListScatter operation |
<T> TensorListScatterIntoExistingList |
tensorListScatterIntoExistingList(Operand<?> inputHandle,
Operand<T> tensor,
Operand<Integer> indices)
Builds an
TensorListScatterIntoExistingList operation |
<T,U extends Number> |
tensorListScatterV2(Operand<T> tensor,
Operand<Integer> indices,
Operand<U> elementShape,
Operand<Integer> numElements)
Builds an
TensorListScatterV2 operation |
<T> TensorListSetItem |
tensorListSetItem(Operand<?> inputHandle,
Operand<Integer> index,
Operand<T> item)
Builds an
TensorListSetItem operation |
<T,U extends Number> |
tensorListSplit(Operand<T> tensor,
Operand<U> elementShape,
Operand<Long> lengths)
Builds an
TensorListSplit operation |
<T> TensorListStack<T> |
tensorListStack(Operand<?> inputHandle,
Operand<Integer> elementShape,
Class<T> elementDtype,
TensorListStack.Options... options)
Builds an
TensorListStack operation |
<T,U extends Number> |
tensorScatterAdd(Operand<T> tensor,
Operand<U> indices,
Operand<T> updates)
Builds an
TensorScatterAdd operation |
<T,U extends Number> |
tensorScatterSub(Operand<T> tensor,
Operand<U> indices,
Operand<T> updates)
Builds an
TensorScatterSub operation |
<T,U extends Number> |
tensorScatterUpdate(Operand<T> tensor,
Operand<U> indices,
Operand<T> updates)
Builds an
TensorScatterUpdate operation |
<T,U extends Number> |
tensorStridedSliceUpdate(Operand<T> input,
Operand<U> begin,
Operand<U> end,
Operand<U> strides,
Operand<T> value,
TensorStridedSliceUpdate.Options... options)
Builds an
TensorStridedSliceUpdate operation |
<T,U extends Number> |
tile(Operand<T> input,
Operand<U> multiples)
Builds an
Tile operation |
Timestamp |
timestamp()
Builds an
Timestamp operation |
TrainOps |
train()
Returns an API for building
train operations |
TryRpc |
tryRpc(Operand<String> address,
Operand<String> method,
Operand<String> request,
TryRpc.Options... options)
Builds an
TryRpc operation |
<T> Unbatch<T> |
unbatch(Operand<T> batchedTensor,
Operand<Long> batchIndex,
Operand<Long> id,
Long timeoutMicros,
Unbatch.Options... options)
Builds an
Unbatch operation |
<T> UnbatchGrad<T> |
unbatchGrad(Operand<T> originalInput,
Operand<Long> batchIndex,
Operand<T> grad,
Operand<Long> id,
UnbatchGrad.Options... options)
Builds an
UnbatchGrad operation |
<T,U extends Number> |
unique(Operand<T> x,
Operand<U> axis)
Builds an
Unique operation |
<T,V extends Number,U extends Number> |
unique(Operand<T> x,
Operand<U> axis,
Class<V> outIdx)
Builds an
Unique operation |
<T,U extends Number> |
uniqueWithCounts(Operand<T> x,
Operand<U> axis)
Builds an
UniqueWithCounts operation |
<T,V extends Number,U extends Number> |
uniqueWithCounts(Operand<T> x,
Operand<U> axis,
Class<V> outIdx)
Builds an
UniqueWithCounts operation |
<T extends Number> |
unravelIndex(Operand<T> indices,
Operand<T> dims)
Builds an
UnravelIndex operation |
<T extends Number,U extends Number> |
unsortedSegmentJoin(Operand<String> inputs,
Operand<T> segmentIds,
Operand<U> numSegments,
UnsortedSegmentJoin.Options... options)
Builds an
UnsortedSegmentJoin operation |
<T> Unstack<T> |
unstack(Operand<T> value,
Long num,
Unstack.Options... options)
Builds an
Unstack operation |
Unstage |
unstage(List<Class<?>> dtypes,
Unstage.Options... options)
Builds an
Unstage operation |
<T> VarHandleOp |
varHandleOp(Class<T> dtype,
Shape shape,
VarHandleOp.Options... options)
Builds an
VarHandleOp operation |
<T> Variable<T> |
variable(Shape shape,
Class<T> dtype,
Variable.Options... options)
Builds an
Variable operation |
VariableShape<Integer> |
variableShape(Operand<?> input)
Builds an
VariableShape operation |
<T extends Number> |
variableShape(Operand<?> input,
Class<T> outType)
Builds an
VariableShape operation |
VarIsInitializedOp |
varIsInitializedOp(Operand<?> resource)
Builds an
VarIsInitializedOp operation |
<T> Where |
where(Operand<T> condition)
Builds an
Where operation |
<T> Where3<T> |
where3(Operand<Boolean> condition,
Operand<T> x,
Operand<T> y)
Builds an
Where3 operation |
Ops |
withControlDependencies(Iterable<Operand<?>> controls)
Returns an API that adds operations to the graph with the provided control dependencies.
|
Ops |
withName(String opName)
Returns an API that uses the provided name for an op.
|
Ops |
withSubScope(String childScopeName)
Returns an API that builds operations with the provided name prefix.
|
<T,U extends Number> |
zeros(Operand<U> dims,
Class<T> type)
Builds an
Zeros operation |
<T> ZerosLike<T> |
zerosLike(Operand<T> x)
Builds an
ZerosLike operation |
public final SummaryOps summary
public final NnOps nn
public final ImageOps image
public final DataOps data
public final IoOps io
public final DtypesOps dtypes
public final LinalgOps linalg
public final RandomOps random
public final StringsOps strings
public final SparseOps sparse
public final BitwiseOps bitwise
public final MathOps math
public final AudioOps audio
public final SignalOps signal
public final TrainOps train
public final QuantizationOps quantization
public <T extends Number> ReduceAll reduceAll(Operand<Boolean> input, Operand<T> axis, ReduceAll.Options... options)
ReduceAll
operationinput
- The tensor to reduce.axis
- The dimensions to reduce. Must be in the rangeoptions
- carries optional attributes valuesReduceAll
public MutexLock mutexLock(Operand<?> mutex)
MutexLock
operationmutex
- The mutex resource to lock.MutexLock
public <T,U extends Number> Sum<T> sum(Operand<T> input, Operand<U> axis, Sum.Options... options)
Sum
operationinput
- The tensor to reduce.axis
- The dimensions to reduce. Must be in the rangeoptions
- carries optional attributes valuesSum
public <T extends Number> Bucketize bucketize(Operand<T> input, List<Float> boundaries)
Bucketize
operationinput
- Any shape of Tensor contains with int or float type.boundaries
- A sorted list of floats gives the boundary of the buckets.Bucketize
public <T> TensorListGather<T> tensorListGather(Operand<?> inputHandle, Operand<Integer> indices, Operand<Integer> elementShape, Class<T> elementDtype)
TensorListGather
operationinputHandle
- indices
- elementShape
- elementDtype
- TensorListGather
public <U,T extends Number> ResourceGather<U> resourceGather(Operand<?> resource, Operand<T> indices, Class<U> dtype, ResourceGather.Options... options)
ResourceGather
operationresource
- indices
- dtype
- options
- carries optional attributes valuesResourceGather
public Constant<Float> constant(long[] shape, FloatBuffer data)
Constant
operationshape
- the tensor shape.data
- a buffer containing the tensor data.IllegalArgumentException
- If the tensor shape is not compatible with the bufferConstant
public <T extends Number,U extends Number> ScatterMin<T> scatterMin(Operand<T> ref, Operand<U> indices, Operand<T> updates, ScatterMin.Options... options)
ScatterMin
operationref
- Should be from a `Variable` node.indices
- A tensor of indices into the first dimension of `ref`.updates
- A tensor of updated values to reduce into `ref`.options
- carries optional attributes valuesScatterMin
public Constant<Float> constant(float data)
Constant
operationdata
- The value to put into the new constant.Constant
public <U extends Number,T extends Number> HistogramFixedWidth<U> histogramFixedWidth(Operand<T> values, Operand<T> valueRange, Operand<Integer> nbins, Class<U> dtype)
HistogramFixedWidth
operationvalues
- Numeric `Tensor`.valueRange
- Shape [2] `Tensor` of same `dtype` as `values`.nbins
- Scalar `int32 Tensor`. Number of histogram bins.dtype
- HistogramFixedWidth
public <T extends Number> HistogramFixedWidth<Integer> histogramFixedWidth(Operand<T> values, Operand<T> valueRange, Operand<Integer> nbins)
HistogramFixedWidth
operationvalues
- Numeric `Tensor`.valueRange
- Shape [2] `Tensor` of same `dtype` as `values`.nbins
- Scalar `int32 Tensor`. Number of histogram bins.HistogramFixedWidth
public Constant<String> constant(byte[] data)
Constant
operationdata
- An array containing the values to put into the new constant. String elements areConstant
public <T> InplaceAdd<T> inplaceAdd(Operand<T> x, Operand<Integer> i, Operand<T> v)
InplaceAdd
operationx
- A `Tensor` of type T.i
- A vector. Indices into the left-most dimension of `x`.v
- A `Tensor` of type T. Same dimension sizes as x except the first dimension, which must be the same as i's size.InplaceAdd
public Constant<Integer> constant(int[][][] data)
Constant
operationdata
- An array containing the values to put into the new constant. The dimensions of theConstant
public <T,U extends Number> Min<T> min(Operand<T> input, Operand<U> axis, Min.Options... options)
Min
operationinput
- The tensor to reduce.axis
- The dimensions to reduce. Must be in the rangeoptions
- carries optional attributes valuesMin
public <T> TensorArraySplit tensorArraySplit(Operand<?> handle, Operand<T> value, Operand<Long> lengths, Operand<Float> flowIn)
TensorArraySplit
operationhandle
- The handle to a TensorArray.value
- The concatenated tensor to write to the TensorArray.lengths
- The vector of lengths, how to split the rows of value into theflowIn
- A float scalar that enforces proper chaining of operations.TensorArraySplit
public <T> VarHandleOp varHandleOp(Class<T> dtype, Shape shape, VarHandleOp.Options... options)
VarHandleOp
operationdtype
- the type of this variable. Must agree with the dtypesshape
- The (possibly partially specified) shape of this variable.options
- carries optional attributes valuesVarHandleOp
public MapIncompleteSize mapIncompleteSize(List<Class<?>> dtypes, MapIncompleteSize.Options... options)
MapIncompleteSize
operationdtypes
- options
- carries optional attributes valuesMapIncompleteSize
public Constant<Boolean> constant(boolean[][][][][][] data)
Constant
operationdata
- An array containing the values to put into the new constant. The dimensions of theConstant
public <T,U extends Number> ScatterDiv<T> scatterDiv(Operand<T> ref, Operand<U> indices, Operand<T> updates, ScatterDiv.Options... options)
ScatterDiv
operationref
- Should be from a `Variable` node.indices
- A tensor of indices into the first dimension of `ref`.updates
- A tensor of values that `ref` is divided by.options
- carries optional attributes valuesScatterDiv
public <T> TensorListConcatLists tensorListConcatLists(Operand<?> inputA, Operand<?> inputB, Class<T> elementDtype)
TensorListConcatLists
operationinputA
- inputB
- elementDtype
- TensorListConcatLists
public Constant<Integer> constant(int[] data)
Constant
operationdata
- An array containing the values to put into the new constant. The dimensions of theConstant
public <T extends Number,U extends Number> FusedBatchNormV3<T,U> fusedBatchNormV3(Operand<T> x, Operand<U> scale, Operand<U> offset, Operand<U> mean, Operand<U> variance, FusedBatchNormV3.Options... options)
FusedBatchNormV3
operationx
- A 4D Tensor for input data.scale
- A 1D Tensor for scaling factor, to scale the normalized x.offset
- A 1D Tensor for offset, to shift to the normalized x.mean
- A 1D Tensor for population mean. Used for inference only;variance
- A 1D Tensor for population variance. Used for inference only;options
- carries optional attributes valuesFusedBatchNormV3
public Unstage unstage(List<Class<?>> dtypes, Unstage.Options... options)
Unstage
operationdtypes
- options
- carries optional attributes valuesUnstage
public <T> TensorArrayWrite tensorArrayWrite(Operand<?> handle, Operand<Integer> index, Operand<T> value, Operand<Float> flowIn)
TensorArrayWrite
operationhandle
- The handle to a TensorArray.index
- The position to write to inside the TensorArray.value
- The tensor to write to the TensorArray.flowIn
- A float scalar that enforces proper chaining of operations.TensorArrayWrite
public <T extends Number,U> ResourceScatterMin resourceScatterMin(Operand<?> resource, Operand<T> indices, Operand<U> updates)
ResourceScatterMin
operationresource
- Should be from a `Variable` node.indices
- A tensor of indices into the first dimension of `ref`.updates
- A tensor of updated values to add to `ref`.ResourceScatterMin
public <T> DestroyTemporaryVariable<T> destroyTemporaryVariable(Operand<T> ref, String varName)
DestroyTemporaryVariable
operationref
- A reference to the temporary variable tensor.varName
- Name of the temporary variable, usually the name of the matchingDestroyTemporaryVariable
public RemoteFusedGraphExecute remoteFusedGraphExecute(Iterable<Operand<?>> inputs, List<Class<?>> Toutputs, String serializedRemoteFusedGraphExecuteInfo)
RemoteFusedGraphExecute
operationinputs
- Arbitrary number of tensors with arbitrary data typesToutputs
- serializedRemoteFusedGraphExecuteInfo
- Serialized protocol bufferRemoteFusedGraphExecute
public <T,U extends Number> Prod<T> prod(Operand<T> input, Operand<U> axis, Prod.Options... options)
Prod
operationinput
- The tensor to reduce.axis
- The dimensions to reduce. Must be in the rangeoptions
- carries optional attributes valuesProd
public <T> Fingerprint fingerprint(Operand<T> data, Operand<String> method)
Fingerprint
operationdata
- Must have rank 1 or higher.method
- Fingerprint method used by this op. Currently available method isFingerprint
public <T,U extends Number> Reverse<T> reverse(Operand<T> tensor, Operand<U> axis)
Reverse
operationtensor
- Up to 8-D.axis
- 1-D. The indices of the dimensions to reverse. Must be in the rangeReverse
public <T> BarrierInsertMany barrierInsertMany(Operand<String> handle, Operand<String> keys, Operand<T> values, Long componentIndex)
BarrierInsertMany
operationhandle
- The handle to a barrier.keys
- A one-dimensional tensor of keys, with length n.values
- An any-dimensional tensor of values, which are associated with thecomponentIndex
- The component of the barrier elements that is being assigned.BarrierInsertMany
public <T> PlaceholderWithDefault<T> placeholderWithDefault(Operand<T> input, Shape shape)
PlaceholderWithDefault
operationinput
- The default value to produce when `output` is not fed.shape
- The (possibly partial) shape of the tensor.PlaceholderWithDefault
public <T extends Number,U> TensorListReserve tensorListReserve(Operand<T> elementShape, Operand<Integer> numElements, Class<U> elementDtype)
TensorListReserve
operationelementShape
- numElements
- elementDtype
- TensorListReserve
public <T,U extends Number> StridedSliceAssign<T> stridedSliceAssign(Operand<T> ref, Operand<U> begin, Operand<U> end, Operand<U> strides, Operand<T> value, StridedSliceAssign.Options... options)
StridedSliceAssign
operationref
- begin
- end
- strides
- value
- options
- carries optional attributes valuesStridedSliceAssign
public <T> MatrixDiagPartV2<T> matrixDiagPartV2(Operand<T> input, Operand<Integer> k, Operand<T> paddingValue)
MatrixDiagPartV2
operationinput
- Rank `r` tensor where `r >= 2`.k
- Diagonal offset(s). Positive value means superdiagonal, 0 refers to the mainpaddingValue
- The value to fill the area outside the specified diagonal band with.MatrixDiagPartV2
public Constant<Float> constant(float[][] data)
Constant
operationdata
- An array containing the values to put into the new constant. The dimensions of theConstant
public <T,U extends Number> EuclideanNorm<T> euclideanNorm(Operand<T> input, Operand<U> axis, EuclideanNorm.Options... options)
EuclideanNorm
operationinput
- The tensor to reduce.axis
- The dimensions to reduce. Must be in the rangeoptions
- carries optional attributes valuesEuclideanNorm
public Constant<Double> constant(long[] shape, DoubleBuffer data)
Constant
operationshape
- the tensor shape.data
- a buffer containing the tensor data.IllegalArgumentException
- If the tensor shape is not compatible with the bufferConstant
public <T> Assign<T> assign(Operand<T> ref, Operand<T> value, Assign.Options... options)
Assign
operationref
- Should be from a `Variable` node. May be uninitialized.value
- The value to be assigned to the variable.options
- carries optional attributes valuesAssign
public <T,U> LookupTableInsert lookupTableInsert(Operand<?> tableHandle, Operand<T> keys, Operand<U> values)
LookupTableInsert
operationtableHandle
- Handle to the table.keys
- Any shape. Keys to look up.values
- Values to associate with keys.LookupTableInsert
public Constant<Float> constant(float[] data)
Constant
operationdata
- An array containing the values to put into the new constant. The dimensions of theConstant
public <T,U extends Number> ReduceSum<T> reduceSum(Operand<T> input, Operand<U> axis, ReduceSum.Options... options)
ReduceSum
operationinput
- The tensor to reduce.axis
- The dimensions to reduce. Must be in the rangeoptions
- carries optional attributes valuesReduceSum
public <U extends Number,T> Size<U> size(Operand<T> input, Class<U> outType)
Size
operationinput
- outType
- Size
public Constant<Boolean> constant(boolean[][][][] data)
Constant
operationdata
- An array containing the values to put into the new constant. The dimensions of theConstant
public <T extends Number,U> ResourceScatterMul resourceScatterMul(Operand<?> resource, Operand<T> indices, Operand<U> updates)
ResourceScatterMul
operationresource
- Should be from a `Variable` node.indices
- A tensor of indices into the first dimension of `ref`.updates
- A tensor of updated values to add to `ref`.ResourceScatterMul
public <U,T extends Number> ResourceGatherNd<U> resourceGatherNd(Operand<?> resource, Operand<T> indices, Class<U> dtype)
ResourceGatherNd
operationresource
- indices
- dtype
- ResourceGatherNd
public <T> GetSessionTensor<T> getSessionTensor(Operand<String> handle, Class<T> dtype)
GetSessionTensor
operationhandle
- The handle for a tensor stored in the session state.dtype
- The type of the output value.GetSessionTensor
public <T> TensorListSetItem tensorListSetItem(Operand<?> inputHandle, Operand<Integer> index, Operand<T> item)
TensorListSetItem
operationinputHandle
- index
- item
- TensorListSetItem
public <T> AssignAdd<T> assignAdd(Operand<T> ref, Operand<T> value, AssignAdd.Options... options)
AssignAdd
operationref
- Should be from a `Variable` node.value
- The value to be added to the variable.options
- carries optional attributes valuesAssignAdd
public <T,U> MutableHashTable mutableHashTable(Class<T> keyDtype, Class<U> valueDtype, MutableHashTable.Options... options)
MutableHashTable
operationkeyDtype
- Type of the table keys.valueDtype
- Type of the table values.options
- carries optional attributes valuesMutableHashTable
public <T> TensorListScatterIntoExistingList tensorListScatterIntoExistingList(Operand<?> inputHandle, Operand<T> tensor, Operand<Integer> indices)
TensorListScatterIntoExistingList
operationinputHandle
- tensor
- indices
- TensorListScatterIntoExistingList
public <T> EditDistance editDistance(Operand<Long> hypothesisIndices, Operand<T> hypothesisValues, Operand<Long> hypothesisShape, Operand<Long> truthIndices, Operand<T> truthValues, Operand<Long> truthShape, EditDistance.Options... options)
EditDistance
operationhypothesisIndices
- The indices of the hypothesis list SparseTensor.hypothesisValues
- The values of the hypothesis list SparseTensor.hypothesisShape
- The shape of the hypothesis list SparseTensor.truthIndices
- The indices of the truth list SparseTensor.truthValues
- The values of the truth list SparseTensor.truthShape
- truth indices, vector.options
- carries optional attributes valuesEditDistance
public <T> ResourceApplyAdamWithAmsgrad resourceApplyAdamWithAmsgrad(Operand<?> var, Operand<?> m, Operand<?> v, Operand<?> vhat, Operand<T> beta1Power, Operand<T> beta2Power, Operand<T> lr, Operand<T> beta1, Operand<T> beta2, Operand<T> epsilon, Operand<T> grad, ResourceApplyAdamWithAmsgrad.Options... options)
ResourceApplyAdamWithAmsgrad
operationvar
- Should be from a Variable().m
- Should be from a Variable().v
- Should be from a Variable().vhat
- Should be from a Variable().beta1Power
- Must be a scalar.beta2Power
- Must be a scalar.lr
- Scaling factor. Must be a scalar.beta1
- Momentum factor. Must be a scalar.beta2
- Momentum factor. Must be a scalar.epsilon
- Ridge term. Must be a scalar.grad
- The gradient.options
- carries optional attributes valuesResourceApplyAdamWithAmsgrad
public <T> Size<Integer> size(Operand<T> input)
Size
operationinput
- Size
public <T,U extends Number> MirrorPad<T> mirrorPad(Operand<T> input, Operand<U> paddings, String mode)
MirrorPad
operationinput
- The input tensor to be padded.paddings
- A two-column matrix specifying the padding sizes. The number ofmode
- Either `REFLECT` or `SYMMETRIC`. In reflect mode the padded regionsMirrorPad
public <T> Constant<T> constant(Object object, Class<T> type)
Constant
operationobject
- a Java object representing the constant.Tensor.create
,
Constant
public OrderedMapPeek orderedMapPeek(Operand<Long> key, Operand<Integer> indices, List<Class<?>> dtypes, OrderedMapPeek.Options... options)
OrderedMapPeek
operationkey
- indices
- dtypes
- options
- carries optional attributes valuesOrderedMapPeek
public Constant<String> constant(byte[][][] data)
Constant
operationdata
- An array containing the values to put into the new constant. String elements areConstant
public <T extends Number,U> ResourceStridedSliceAssign resourceStridedSliceAssign(Operand<?> ref, Operand<T> begin, Operand<T> end, Operand<T> strides, Operand<U> value, ResourceStridedSliceAssign.Options... options)
ResourceStridedSliceAssign
operationref
- begin
- end
- strides
- value
- options
- carries optional attributes valuesResourceStridedSliceAssign
public Constant<Integer> constant(int[][][][] data)
Constant
operationdata
- An array containing the values to put into the new constant. The dimensions of theConstant
public <T> TensorListStack<T> tensorListStack(Operand<?> inputHandle, Operand<Integer> elementShape, Class<T> elementDtype, TensorListStack.Options... options)
TensorListStack
operationinputHandle
- elementShape
- elementDtype
- options
- carries optional attributes valuesTensorListStack
public <T> StatefulStandardNormal<Float> statefulStandardNormal(Operand<?> resource, Operand<T> shape)
StatefulStandardNormal
operationresource
- The handle of the resource variable that stores the state of the RNG.shape
- The shape of the output tensor.StatefulStandardNormal
public Constant<Double> constant(double[] data)
Constant
operationdata
- An array containing the values to put into the new constant. The dimensions of theConstant
public <T> Lu<T,Integer> lu(Operand<T> input)
Lu
operationinput
- A tensor of shape `[..., M, M]` whose inner-most 2 dimensions form matrices ofLu
public <U,T extends Number> OneHot<U> oneHot(Operand<T> indices, Operand<Integer> depth, Operand<U> onValue, Operand<U> offValue, OneHot.Options... options)
OneHot
operationindices
- A tensor of indices.depth
- A scalar defining the depth of the one hot dimension.onValue
- A scalar defining the value to fill in output when `indices[j] = i`.offValue
- A scalar defining the value to fill in output when `indices[j] != i`.options
- carries optional attributes valuesOneHot
public DeleteSessionTensor deleteSessionTensor(Operand<String> handle)
DeleteSessionTensor
operationhandle
- The handle for a tensor stored in the session state.DeleteSessionTensor
public Gradients gradients(Operand<?> y, Iterable<? extends Operand<?>> x, Gradients.Options... options)
Gradients
operationy
- output of the function to derivex
- inputs of the function for which partial derivatives are computedoptions
- carries optional attributes valuesGradients
IllegalArgumentException
- if execution environment is not a graphGradients
public <T extends Number,U extends Number> UnsortedSegmentJoin unsortedSegmentJoin(Operand<String> inputs, Operand<T> segmentIds, Operand<U> numSegments, UnsortedSegmentJoin.Options... options)
UnsortedSegmentJoin
operationinputs
- The input to be joined.segmentIds
- A tensor whose shape is a prefix of data.shape. Negative segment ids are notnumSegments
- A scalar.options
- carries optional attributes valuesUnsortedSegmentJoin
public <T extends Number,U> ResourceScatterNdSub resourceScatterNdSub(Operand<?> ref, Operand<T> indices, Operand<U> updates, ResourceScatterNdSub.Options... options)
ResourceScatterNdSub
operationref
- A resource handle. Must be from a VarHandleOp.indices
- A Tensor. Must be one of the following types: int32, int64.updates
- A Tensor. Must have the same type as ref. A tensor ofoptions
- carries optional attributes valuesResourceScatterNdSub
public <T extends Number,U> ResourceScatterNdUpdate resourceScatterNdUpdate(Operand<?> ref, Operand<T> indices, Operand<U> updates, ResourceScatterNdUpdate.Options... options)
ResourceScatterNdUpdate
operationref
- A resource handle. Must be from a VarHandleOp.indices
- A Tensor. Must be one of the following types: int32, int64.updates
- A Tensor. Must have the same type as ref. A tensor of updatedoptions
- carries optional attributes valuesResourceScatterNdUpdate
public Constant<Float> constant(float[][][][][][] data)
Constant
operationdata
- An array containing the values to put into the new constant. The dimensions of theConstant
public <T extends Number,U extends Number> LinSpace<T> linSpace(Operand<T> start, Operand<T> stop, Operand<U> num)
LinSpace
operationstart
- 0-D tensor. First entry in the range.stop
- 0-D tensor. Last entry in the range.num
- 0-D tensor. Number of values to generate.LinSpace
public CombinedNonMaxSuppression combinedNonMaxSuppression(Operand<Float> boxes, Operand<Float> scores, Operand<Integer> maxOutputSizePerClass, Operand<Integer> maxTotalSize, Operand<Float> iouThreshold, Operand<Float> scoreThreshold, CombinedNonMaxSuppression.Options... options)
CombinedNonMaxSuppression
operationboxes
- A 4-D float tensor of shape `[batch_size, num_boxes, q, 4]`. If `q` is 1 thenscores
- A 3-D float tensor of shape `[batch_size, num_boxes, num_classes]`maxOutputSizePerClass
- A scalar integer tensor representing the maximum number ofmaxTotalSize
- A scalar representing maximum number of boxes retained over all classes.iouThreshold
- A 0-D float tensor representing the threshold for deciding whetherscoreThreshold
- A 0-D float tensor representing the threshold for deciding when to removeoptions
- carries optional attributes valuesCombinedNonMaxSuppression
public <T extends Number> StringNGrams<T> stringNGrams(Operand<String> data, Operand<T> dataSplits, String separator, List<Long> ngramWidths, String leftPad, String rightPad, Long padWidth, Boolean preserveShortSequences)
StringNGrams
operationdata
- The values tensor of the ragged string tensor to make ngrams out of. Must be adataSplits
- The splits tensor of the ragged string tensor to make ngrams out of.separator
- The string to append between elements of the token. Use "" for no separator.ngramWidths
- The sizes of the ngrams to create.leftPad
- The string to use to pad the left side of the ngram sequence. Only used ifrightPad
- The string to use to pad the right side of the ngram sequence. Only used ifpadWidth
- The number of padding elements to add to each side of eachpreserveShortSequences
- StringNGrams
public <T> ZerosLike<T> zerosLike(Operand<T> x)
ZerosLike
operationx
- a tensor of type T.ZerosLike
public <T extends Number,U> ResourceScatterSub resourceScatterSub(Operand<?> resource, Operand<T> indices, Operand<U> updates)
ResourceScatterSub
operationresource
- Should be from a `Variable` node.indices
- A tensor of indices into the first dimension of `ref`.updates
- A tensor of updated values to add to `ref`.ResourceScatterSub
public <T> InplaceSub<T> inplaceSub(Operand<T> x, Operand<Integer> i, Operand<T> v)
InplaceSub
operationx
- A `Tensor` of type T.i
- A vector. Indices into the left-most dimension of `x`.v
- A `Tensor` of type T. Same dimension sizes as x except the first dimension, which must be the same as i's size.InplaceSub
public <T,U extends Number> ReduceMin<T> reduceMin(Operand<T> input, Operand<U> axis, ReduceMin.Options... options)
ReduceMin
operationinput
- The tensor to reduce.axis
- The dimensions to reduce. Must be in the rangeoptions
- carries optional attributes valuesReduceMin
public Constant<Double> constant(double[][] data)
Constant
operationdata
- An array containing the values to put into the new constant. The dimensions of theConstant
public <U,T> LookupTableFind<U> lookupTableFind(Operand<?> tableHandle, Operand<T> keys, Operand<U> defaultValue)
LookupTableFind
operationtableHandle
- Handle to the table.keys
- Any shape. Keys to look up.defaultValue
- LookupTableFind
public StringUpper stringUpper(Operand<String> input, StringUpper.Options... options)
StringUpper
operationinput
- options
- carries optional attributes valuesStringUpper
public <T> OnesLike<T> onesLike(Operand<T> x)
OnesLike
operationx
- a tensor of type T.OnesLike
public <T,U extends Number> ReduceMax<T> reduceMax(Operand<T> input, Operand<U> axis, ReduceMax.Options... options)
ReduceMax
operationinput
- The tensor to reduce.axis
- The dimensions to reduce. Must be in the rangeoptions
- carries optional attributes valuesReduceMax
public <T> DynamicPartition<T> dynamicPartition(Operand<T> data, Operand<Integer> partitions, Long numPartitions)
DynamicPartition
operationdata
- partitions
- Any shape. Indices in the range `[0, num_partitions)`.numPartitions
- The number of partitions to output.DynamicPartition
public <T extends Number> All all(Operand<Boolean> input, Operand<T> axis, All.Options... options)
All
operationinput
- The tensor to reduce.axis
- The dimensions to reduce. Must be in the rangeoptions
- carries optional attributes valuesAll
public <T,U extends Number> BatchToSpace<T> batchToSpace(Operand<T> input, Operand<U> crops, Long blockSize)
BatchToSpace
operationinput
- 4-D tensor with shapecrops
- 2-D tensor of non-negative integers with shape `[2, 2]`. It specifiesblockSize
- BatchToSpace
public <T,U extends Number> ScatterNdSub<T> scatterNdSub(Operand<T> ref, Operand<U> indices, Operand<T> updates, ScatterNdSub.Options... options)
ScatterNdSub
operationref
- A mutable Tensor. Should be from a Variable node.indices
- A Tensor. Must be one of the following types: int32, int64.updates
- A Tensor. Must have the same type as ref. A tensor of updated valuesoptions
- carries optional attributes valuesScatterNdSub
public <T,U> HashTable hashTable(Class<T> keyDtype, Class<U> valueDtype, HashTable.Options... options)
HashTable
operationkeyDtype
- Type of the table keys.valueDtype
- Type of the table values.options
- carries optional attributes valuesHashTable
public IdentityN identityN(Iterable<Operand<?>> input)
IdentityN
operationinput
- IdentityN
public <T> ClipByValue<T> clipByValue(Operand<T> t, Operand<T> clipValueMin, Operand<T> clipValueMax)
ClipByValue
operationt
- A `Tensor`.clipValueMin
- A 0-D (scalar) `Tensor`, or a `Tensor` with the same shapeclipValueMax
- A 0-D (scalar) `Tensor`, or a `Tensor` with the same shapeClipByValue
public <T> TensorListPushBackBatch tensorListPushBackBatch(Operand<?> inputHandles, Operand<T> tensor)
TensorListPushBackBatch
operationinputHandles
- tensor
- TensorListPushBackBatch
public Constant<Long> constant(long[][][] data)
Constant
operationdata
- An array containing the values to put into the new constant. The dimensions of theConstant
public <T> RefNextIteration<T> refNextIteration(Operand<T> data)
RefNextIteration
operationdata
- The tensor to be made available to the next iteration.RefNextIteration
public <T,U extends Number> Unique<T,Integer> unique(Operand<T> x, Operand<U> axis)
Unique
operationx
- A `Tensor`.axis
- A `Tensor` of type `int32` (default: None). The axis of the Tensor toUnique
public <T> TensorArrayGather<T> tensorArrayGather(Operand<?> handle, Operand<Integer> indices, Operand<Float> flowIn, Class<T> dtype, TensorArrayGather.Options... options)
TensorArrayGather
operationhandle
- The handle to a TensorArray.indices
- The locations in the TensorArray from which to read tensor elements.flowIn
- A float scalar that enforces proper chaining of operations.dtype
- The type of the elem that is returned.options
- carries optional attributes valuesTensorArrayGather
public <T,U extends Number> TensorListScatterV2 tensorListScatterV2(Operand<T> tensor, Operand<Integer> indices, Operand<U> elementShape, Operand<Integer> numElements)
TensorListScatterV2
operationtensor
- indices
- elementShape
- numElements
- TensorListScatterV2
public Timestamp timestamp()
Timestamp
operationTimestamp
public <T,U extends Number> Slice<T> slice(Operand<T> input, Operand<U> begin, Operand<U> size)
Slice
operationinput
- begin
- begin[i] specifies the offset into the 'i'th dimension ofsize
- size[i] specifies the number of elements of the 'i'th dimensionSlice
public MapSize mapSize(List<Class<?>> dtypes, MapSize.Options... options)
MapSize
operationdtypes
- options
- carries optional attributes valuesMapSize
public <T,V extends Number,U extends Number> UniqueWithCounts<T,V> uniqueWithCounts(Operand<T> x, Operand<U> axis, Class<V> outIdx)
UniqueWithCounts
operationx
- A `Tensor`.axis
- A `Tensor` of type `int32` (default: None). The axis of the Tensor tooutIdx
- UniqueWithCounts
public Constant<Boolean> constant(boolean[][][][][] data)
Constant
operationdata
- An array containing the values to put into the new constant. The dimensions of theConstant
public <T> TensorListConcat<T> tensorListConcat(Operand<?> inputHandle, Class<T> elementDtype, TensorListConcat.Options... options)
TensorListConcat
operationinputHandle
- elementDtype
- options
- carries optional attributes valuesTensorListConcat
public <T extends Number,U> ResourceScatterUpdate resourceScatterUpdate(Operand<?> resource, Operand<T> indices, Operand<U> updates)
ResourceScatterUpdate
operationresource
- Should be from a `Variable` node.indices
- A tensor of indices into the first dimension of `ref`.updates
- A tensor of updated values to add to `ref`.ResourceScatterUpdate
public <T> Einsum<T> einsum(Iterable<Operand<T>> inputs, String equation)
Einsum
operationinputs
- List of 1 or 2 Tensors.equation
- String describing the Einstein Summation operation; in the format of np.einsum.Einsum
public <T extends Number> CudnnRNNCanonicalToParamsV2<T> cudnnRNNCanonicalToParamsV2(Operand<Integer> numLayers, Operand<Integer> numUnits, Operand<Integer> inputSize, Iterable<Operand<T>> weights, Iterable<Operand<T>> biases, CudnnRNNCanonicalToParamsV2.Options... options)
CudnnRNNCanonicalToParamsV2
operationnumLayers
- numUnits
- inputSize
- weights
- biases
- options
- carries optional attributes valuesCudnnRNNCanonicalToParamsV2
public <T extends Number,U> ResourceScatterMax resourceScatterMax(Operand<?> resource, Operand<T> indices, Operand<U> updates)
ResourceScatterMax
operationresource
- Should be from a `Variable` node.indices
- A tensor of indices into the first dimension of `ref`.updates
- A tensor of updated values to add to `ref`.ResourceScatterMax
public <T> MatrixSetDiagV2<T> matrixSetDiagV2(Operand<T> input, Operand<T> diagonal, Operand<Integer> k)
MatrixSetDiagV2
operationinput
- Rank `r+1`, where `r >= 1`.diagonal
- Rank `r` when `k` is an integer or `k[0] == k[1]`. Otherwise, it has rank `r+1`.k
- Diagonal offset(s). Positive value means superdiagonal, 0 refers to the mainMatrixSetDiagV2
public <V extends Number,T extends Number,U extends Number> StatefulRandomBinomial<V> statefulRandomBinomial(Operand<?> resource, Operand<Long> algorithm, Operand<T> shape, Operand<U> counts, Operand<U> probs, Class<V> dtype)
StatefulRandomBinomial
operationresource
- algorithm
- shape
- counts
- probs
- dtype
- StatefulRandomBinomial
public Rpc rpc(Operand<String> address, Operand<String> method, Operand<String> request, Rpc.Options... options)
Rpc
operationaddress
- `0-D` or `1-D`. The address (i.e. host_name:port) of the RPC server.method
- `0-D` or `1-D`. The method address on the RPC server.request
- `0-D` or `1-D`. Serialized proto strings: the rpc request argument.options
- carries optional attributes valuesRpc
public Constant<Integer> constant(int[][][][][][] data)
Constant
operationdata
- An array containing the values to put into the new constant. The dimensions of theConstant
public <T,U extends Number> SplitV<T> splitV(Operand<T> value, Operand<U> sizeSplits, Operand<Integer> axis, Long numSplit)
SplitV
operationvalue
- The tensor to split.sizeSplits
- list containing the sizes of each output tensor along the splitaxis
- 0-D. The dimension along which to split. Must be in the rangenumSplit
- SplitV
public Batch batch(Iterable<Operand<?>> inTensors, Long numBatchThreads, Long maxBatchSize, Long batchTimeoutMicros, Long gradTimeoutMicros, Batch.Options... options)
Batch
operationinTensors
- numBatchThreads
- maxBatchSize
- batchTimeoutMicros
- gradTimeoutMicros
- options
- carries optional attributes valuesBatch
public <T> TensorArrayRead<T> tensorArrayRead(Operand<?> handle, Operand<Integer> index, Operand<Float> flowIn, Class<T> dtype)
TensorArrayRead
operationhandle
- The handle to a TensorArray.index
- flowIn
- A float scalar that enforces proper chaining of operations.dtype
- The type of the elem that is returned.TensorArrayRead
public <T,U extends Number> ScatterMul<T> scatterMul(Operand<T> ref, Operand<U> indices, Operand<T> updates, ScatterMul.Options... options)
ScatterMul
operationref
- Should be from a `Variable` node.indices
- A tensor of indices into the first dimension of `ref`.updates
- A tensor of updated values to multiply to `ref`.options
- carries optional attributes valuesScatterMul
public <T> IsVariableInitialized isVariableInitialized(Operand<T> ref)
IsVariableInitialized
operationref
- Should be from a `Variable` node. May be uninitialized.IsVariableInitialized
public <T extends Number,U> ResourceScatterAdd resourceScatterAdd(Operand<?> resource, Operand<T> indices, Operand<U> updates)
ResourceScatterAdd
operationresource
- Should be from a `Variable` node.indices
- A tensor of indices into the first dimension of `ref`.updates
- A tensor of updated values to add to `ref`.ResourceScatterAdd
public <U,T> StatefulStandardNormal<U> statefulStandardNormal(Operand<?> resource, Operand<T> shape, Class<U> dtype)
StatefulStandardNormal
operationresource
- The handle of the resource variable that stores the state of the RNG.shape
- The shape of the output tensor.dtype
- The type of the output.StatefulStandardNormal
public <T> Empty<T> empty(Operand<Integer> shape, Class<T> dtype, Empty.Options... options)
Empty
operationshape
- 1-D. Represents the shape of the output tensor.dtype
- options
- carries optional attributes valuesEmpty
public <T> ParallelConcat<T> parallelConcat(Iterable<Operand<T>> values, Shape shape)
ParallelConcat
operationvalues
- Tensors to be concatenated. All must have size 1 in the first dimensionshape
- the final shape of the result; should be equal to the shapes of any inputParallelConcat
public Constant<Long> constant(long[] data)
Constant
operationdata
- An array containing the values to put into the new constant. The dimensions of theConstant
public BarrierIncompleteSize barrierIncompleteSize(Operand<String> handle)
BarrierIncompleteSize
operationhandle
- The handle to a barrier.BarrierIncompleteSize
public <T,U extends Number> TensorScatterUpdate<T> tensorScatterUpdate(Operand<T> tensor, Operand<U> indices, Operand<T> updates)
TensorScatterUpdate
operationtensor
- Tensor to copy/update.indices
- Index tensor.updates
- Updates to scatter into output.TensorScatterUpdate
public Constant<Long> constant(long[][][][][] data)
Constant
operationdata
- An array containing the values to put into the new constant. The dimensions of theConstant
public <T> Where3<T> where3(Operand<Boolean> condition, Operand<T> x, Operand<T> y)
Where3
operationcondition
- x
- = A `Tensor` which may have the same shape as `condition`.y
- = A `Tensor` with the same type and shape as `x`.Where3
public Stage stage(Iterable<Operand<?>> values, Stage.Options... options)
Stage
operationvalues
- a list of tensorsoptions
- carries optional attributes valuesStage
public <T> TensorListPopBack<T> tensorListPopBack(Operand<?> inputHandle, Operand<Integer> elementShape, Class<T> elementDtype)
TensorListPopBack
operationinputHandle
- elementShape
- elementDtype
- TensorListPopBack
public <T,U extends Number> TensorListFromTensor tensorListFromTensor(Operand<T> tensor, Operand<U> elementShape)
TensorListFromTensor
operationtensor
- elementShape
- TensorListFromTensor
public OrderedMapIncompleteSize orderedMapIncompleteSize(List<Class<?>> dtypes, OrderedMapIncompleteSize.Options... options)
OrderedMapIncompleteSize
operationdtypes
- options
- carries optional attributes valuesOrderedMapIncompleteSize
public TensorListLength tensorListLength(Operand<?> inputHandle)
TensorListLength
operationinputHandle
- TensorListLength
public Constant<Double> constant(double[][][] data)
Constant
operationdata
- An array containing the values to put into the new constant. The dimensions of theConstant
public <T> Constant<T> constant(Class<T> type, long[] shape, ByteBuffer data)
Constant
operationtype
- the tensor datatype.shape
- the tensor shape.data
- a buffer containing the tensor data.IllegalArgumentException
- If the tensor datatype or shape is not compatible with theConstant
public <T extends Number,U extends Number> FusedBatchNormGradV3<T,U> fusedBatchNormGradV3(Operand<T> yBackprop, Operand<T> x, Operand<Float> scale, Operand<U> reserveSpace1, Operand<U> reserveSpace2, Operand<U> reserveSpace3, FusedBatchNormGradV3.Options... options)
FusedBatchNormGradV3
operationyBackprop
- A 4D Tensor for the gradient with respect to y.x
- A 4D Tensor for input data.scale
- A 1D Tensor for scaling factor, to scale the normalized x.reserveSpace1
- When is_training is True, a 1D Tensor for the computed batchreserveSpace2
- When is_training is True, a 1D Tensor for the computed batchreserveSpace3
- When is_training is True, a 1D Tensor for some intermediate results to be reusedoptions
- carries optional attributes valuesFusedBatchNormGradV3
public <T,U extends Number> UniqueWithCounts<T,Integer> uniqueWithCounts(Operand<T> x, Operand<U> axis)
UniqueWithCounts
operationx
- A `Tensor`.axis
- A `Tensor` of type `int32` (default: None). The axis of the Tensor toUniqueWithCounts
public <U extends Number,T> Shape<U> shape(Operand<T> input, Class<U> outType)
Shape
operationinput
- outType
- Shape
public <U,T extends Number> TensorListConcatV2<U> tensorListConcatV2(Operand<?> inputHandle, Operand<T> elementShape, Operand<Long> leadingDims, Class<U> elementDtype)
TensorListConcatV2
operationinputHandle
- elementShape
- leadingDims
- elementDtype
- TensorListConcatV2
public <T> Unstack<T> unstack(Operand<T> value, Long num, Unstack.Options... options)
Unstack
operationvalue
- 1-D or higher, with `axis` dimension size equal to `num`.num
- options
- carries optional attributes valuesUnstack
public <U,T extends Number> Fill<U> fill(Operand<T> dims, Operand<U> value)
Fill
operationdims
- 1-D. Represents the shape of the output tensor.value
- 0-D (scalar). Value to fill the returned tensor.Fill
public <T,U extends Number> ReverseSequence<T> reverseSequence(Operand<T> input, Operand<U> seqLengths, Long seqDim, ReverseSequence.Options... options)
ReverseSequence
operationinput
- The input to reverse.seqLengths
- 1-D with length `input.dims(batch_dim)` andseqDim
- The dimension which is partially reversed.options
- carries optional attributes valuesReverseSequence
public <T> Unbatch<T> unbatch(Operand<T> batchedTensor, Operand<Long> batchIndex, Operand<Long> id, Long timeoutMicros, Unbatch.Options... options)
Unbatch
operationbatchedTensor
- batchIndex
- id
- timeoutMicros
- options
- carries optional attributes valuesUnbatch
public TensorArrayGradWithShape tensorArrayGradWithShape(Operand<?> handle, Operand<Float> flowIn, Operand<Integer> shapeToPrepend, String source)
TensorArrayGradWithShape
operationhandle
- The handle to the forward TensorArray.flowIn
- A float scalar that enforces proper chaining of operations.shapeToPrepend
- An int32 vector representing a shape. Elements in the gradient accumulator willsource
- The gradient source string, used to decide which gradient TensorArrayTensorArrayGradWithShape
public <T,U extends Number> QuantizedConcatV2<T> quantizedConcatV2(Iterable<Operand<T>> values, Operand<U> axis, Iterable<Operand<Float>> inputMins, Iterable<Operand<Float>> inputMaxes)
QuantizedConcatV2
operationvalues
- axis
- inputMins
- inputMaxes
- QuantizedConcatV2
public <T> TemporaryVariable<T> temporaryVariable(Shape shape, Class<T> dtype, TemporaryVariable.Options... options)
TemporaryVariable
operationshape
- The shape of the variable tensor.dtype
- The type of elements in the variable tensor.options
- carries optional attributes valuesTemporaryVariable
public <T> StopGradient<T> stopGradient(Operand<T> input)
StopGradient
operationinput
- StopGradient
public <T> Identity<T> identity(Operand<T> input)
Identity
operationinput
- Identity
public Constant<Long> constant(long[][][][] data)
Constant
operationdata
- An array containing the values to put into the new constant. The dimensions of theConstant
public <T extends Number> BroadcastDynamicShape<T> broadcastDynamicShape(Operand<T> s0, Operand<T> s1)
BroadcastDynamicShape
operations0
- s1
- BroadcastDynamicShape
public TensorArrayClose tensorArrayClose(Operand<?> handle)
TensorArrayClose
operationhandle
- The handle to a TensorArray (output of TensorArray or TensorArrayGrad).TensorArrayClose
public Skipgram skipgram(String filename, Long batchSize, Skipgram.Options... options)
Skipgram
operationfilename
- The corpus's text file name.batchSize
- The size of produced batch.options
- carries optional attributes valuesSkipgram
public <T,U extends Number> TensorScatterAdd<T> tensorScatterAdd(Operand<T> tensor, Operand<U> indices, Operand<T> updates)
TensorScatterAdd
operationtensor
- Tensor to copy/update.indices
- Index tensor.updates
- Updates to scatter into output.TensorScatterAdd
public Constant<String> constant(String data)
Constant
operationdata
- The string to put into the new constant.Constant
public <T extends Number> CudnnRNNParamsToCanonicalV2<T> cudnnRNNParamsToCanonicalV2(Operand<Integer> numLayers, Operand<Integer> numUnits, Operand<Integer> inputSize, Operand<T> params, Long numParamsWeights, Long numParamsBiases, CudnnRNNParamsToCanonicalV2.Options... options)
CudnnRNNParamsToCanonicalV2
operationnumLayers
- numUnits
- inputSize
- params
- numParamsWeights
- numParamsBiases
- options
- carries optional attributes valuesCudnnRNNParamsToCanonicalV2
public Constant<String> constant(byte[][][][][][] data)
Constant
operationdata
- An array containing the values to put into the new constant. String elements areConstant
public <T> SetSize setSize(Operand<Long> setIndices, Operand<T> setValues, Operand<Long> setShape, SetSize.Options... options)
SetSize
operationsetIndices
- 2D `Tensor`, indices of a `SparseTensor`.setValues
- 1D `Tensor`, values of a `SparseTensor`.setShape
- 1D `Tensor`, shape of a `SparseTensor`.options
- carries optional attributes valuesSetSize
public <T,U extends Number> ScatterNdNonAliasingAdd<T> scatterNdNonAliasingAdd(Operand<T> input, Operand<U> indices, Operand<T> updates)
ScatterNdNonAliasingAdd
operationinput
- A Tensor.indices
- A Tensor. Must be one of the following types: `int32`, `int64`.updates
- A Tensor. Must have the same type as ref. A tensor of updated valuesScatterNdNonAliasingAdd
public <T,U extends Number> ScatterAdd<T> scatterAdd(Operand<T> ref, Operand<U> indices, Operand<T> updates, ScatterAdd.Options... options)
ScatterAdd
operationref
- Should be from a `Variable` node.indices
- A tensor of indices into the first dimension of `ref`.updates
- A tensor of updated values to add to `ref`.options
- carries optional attributes valuesScatterAdd
public <U,T extends Number> StridedSliceGrad<U> stridedSliceGrad(Operand<T> shape, Operand<T> begin, Operand<T> end, Operand<T> strides, Operand<U> dy, StridedSliceGrad.Options... options)
StridedSliceGrad
operationshape
- begin
- end
- strides
- dy
- options
- carries optional attributes valuesStridedSliceGrad
public <T> AssignSub<T> assignSub(Operand<T> ref, Operand<T> value, AssignSub.Options... options)
AssignSub
operationref
- Should be from a `Variable` node.value
- The value to be subtracted to the variable.options
- carries optional attributes valuesAssignSub
public LoopCond loopCond(Operand<Boolean> input)
LoopCond
operationinput
- A boolean scalar, representing the branch predicate of the Switch op.LoopCond
public MapUnstage mapUnstage(Operand<Long> key, Operand<Integer> indices, List<Class<?>> dtypes, MapUnstage.Options... options)
MapUnstage
operationkey
- indices
- dtypes
- options
- carries optional attributes valuesMapUnstage
public <T,U extends Number> TensorListSplit tensorListSplit(Operand<T> tensor, Operand<U> elementShape, Operand<Long> lengths)
TensorListSplit
operationtensor
- elementShape
- lengths
- TensorListSplit
public Mutex mutex(Mutex.Options... options)
Mutex
operationoptions
- carries optional attributes valuesMutex
public <T,U extends Number> ResourceSparseApplyKerasMomentum resourceSparseApplyKerasMomentum(Operand<?> var, Operand<?> accum, Operand<T> lr, Operand<T> grad, Operand<U> indices, Operand<T> momentum, ResourceSparseApplyKerasMomentum.Options... options)
ResourceSparseApplyKerasMomentum
operationvar
- Should be from a Variable().accum
- Should be from a Variable().lr
- Learning rate. Must be a scalar.grad
- The gradient.indices
- A vector of indices into the first dimension of var and accum.momentum
- Momentum. Must be a scalar.options
- carries optional attributes valuesResourceSparseApplyKerasMomentum
public <T> MatrixDiagV2<T> matrixDiagV2(Operand<T> diagonal, Operand<Integer> k, Operand<Integer> numRows, Operand<Integer> numCols, Operand<T> paddingValue)
MatrixDiagV2
operationdiagonal
- Rank `r`, where `r >= 1`k
- Diagonal offset(s). Positive value means superdiagonal, 0 refers to the mainnumRows
- The number of rows of the output matrix. If it is not provided, the op assumesnumCols
- The number of columns of the output matrix. If it is not provided, the oppaddingValue
- The number to fill the area outside the specified diagonal band with.MatrixDiagV2
public Constant<Boolean> constant(boolean[] data)
Constant
operationdata
- An array containing the values to put into the new constant. The dimensions of theConstant
public <T extends Number,U> ResourceScatterDiv resourceScatterDiv(Operand<?> resource, Operand<T> indices, Operand<U> updates)
ResourceScatterDiv
operationresource
- Should be from a `Variable` node.indices
- A tensor of indices into the first dimension of `ref`.updates
- A tensor of updated values to add to `ref`.ResourceScatterDiv
public <T> TensorArrayScatter tensorArrayScatter(Operand<?> handle, Operand<Integer> indices, Operand<T> value, Operand<Float> flowIn)
TensorArrayScatter
operationhandle
- The handle to a TensorArray.indices
- The locations at which to write the tensor elements.value
- The concatenated tensor to write to the TensorArray.flowIn
- A float scalar that enforces proper chaining of operations.TensorArrayScatter
public Constant<Double> constant(double[][][][][][] data)
Constant
operationdata
- An array containing the values to put into the new constant. The dimensions of theConstant
public Constant<String> constant(byte[][][][][] data)
Constant
operationdata
- An array containing the values to put into the new constant. String elements areConstant
public Constant<Integer> constant(long[] shape, IntBuffer data)
Constant
operationshape
- the tensor shape.data
- a buffer containing the tensor data.IllegalArgumentException
- If the tensor shape is not compatible with the bufferConstant
public Gradients gradients(Iterable<? extends Operand<?>> y, Iterable<? extends Operand<?>> x, Gradients.Options... options)
Gradients
operationy
- outputs of the function to derivex
- inputs of the function for which partial derivatives are computedoptions
- carries optional attributes valuesGradients
IllegalArgumentException
- if execution environment is not a graphGradients
public <T,U> MutableDenseHashTable mutableDenseHashTable(Operand<T> emptyKey, Operand<T> deletedKey, Class<U> valueDtype, MutableDenseHashTable.Options... options)
MutableDenseHashTable
operationemptyKey
- The key used to represent empty key buckets internally. Must notdeletedKey
- valueDtype
- Type of the table values.options
- carries optional attributes valuesMutableDenseHashTable
public InitializeTableFromTextFile initializeTableFromTextFile(Operand<?> tableHandle, Operand<String> filename, Long keyIndex, Long valueIndex, InitializeTableFromTextFile.Options... options)
InitializeTableFromTextFile
operationtableHandle
- Handle to a table which will be initialized.filename
- Filename of a vocabulary text file.keyIndex
- Column index in a line to get the table `key` values from.valueIndex
- Column index that represents information of a line to get the tableoptions
- carries optional attributes valuesInitializeTableFromTextFile
public <T> TensorListPushBack tensorListPushBack(Operand<?> inputHandle, Operand<T> tensor)
TensorListPushBack
operationinputHandle
- tensor
- TensorListPushBack
public <T,U extends Number> TensorStridedSliceUpdate<T> tensorStridedSliceUpdate(Operand<T> input, Operand<U> begin, Operand<U> end, Operand<U> strides, Operand<T> value, TensorStridedSliceUpdate.Options... options)
TensorStridedSliceUpdate
operationinput
- begin
- end
- strides
- value
- options
- carries optional attributes valuesTensorStridedSliceUpdate
public <T extends Number> Range<T> range(Operand<T> start, Operand<T> limit, Operand<T> delta)
Range
operationstart
- 0-D (scalar). First entry in the sequence.limit
- 0-D (scalar). Upper limit of sequence, exclusive.delta
- 0-D (scalar). Optional. Default is 1. Number that increments `start`.Range
public <T,U> InitializeTable initializeTable(Operand<?> tableHandle, Operand<T> keys, Operand<U> values)
InitializeTable
operationtableHandle
- Handle to a table which will be initialized.keys
- Keys of type Tkey.values
- Values of type Tval.InitializeTable
public <T> DynamicStitch<T> dynamicStitch(Iterable<Operand<Integer>> indices, Iterable<Operand<T>> data)
DynamicStitch
operationindices
- data
- DynamicStitch
public <T,U extends Number> ScatterSub<T> scatterSub(Operand<T> ref, Operand<U> indices, Operand<T> updates, ScatterSub.Options... options)
ScatterSub
operationref
- Should be from a `Variable` node.indices
- A tensor of indices into the first dimension of `ref`.updates
- A tensor of updated values to subtract from `ref`.options
- carries optional attributes valuesScatterSub
public <T extends Number> DrawBoundingBoxesV2<T> drawBoundingBoxesV2(Operand<T> images, Operand<Float> boxes, Operand<Float> colors)
DrawBoundingBoxesV2
operationimages
- 4-D with shape `[batch, height, width, depth]`. A batch of images.boxes
- 3-D with shape `[batch, num_bounding_boxes, 4]` containing boundingcolors
- 2-D. A list of RGBA colors to cycle through for the boxes.DrawBoundingBoxesV2
public TensorListResize tensorListResize(Operand<?> inputHandle, Operand<Integer> size)
TensorListResize
operationinputHandle
- size
- TensorListResize
public <T,U extends Number> SetDiff1d<T,U> setDiff1d(Operand<T> x, Operand<T> y, Class<U> outIdx)
SetDiff1d
operationx
- 1-D. Values to keep.y
- 1-D. Values to remove.outIdx
- SetDiff1d
public <T,U extends Number> TensorListScatter tensorListScatter(Operand<T> tensor, Operand<Integer> indices, Operand<U> elementShape)
TensorListScatter
operationtensor
- indices
- elementShape
- TensorListScatter
public Constant<Boolean> constant(boolean[][] data)
Constant
operationdata
- An array containing the values to put into the new constant. The dimensions of theConstant
public <T,U> LookupTableImport lookupTableImport(Operand<?> tableHandle, Operand<T> keys, Operand<U> values)
LookupTableImport
operationtableHandle
- Handle to the table.keys
- Any shape. Keys to look up.values
- Values to associate with keys.LookupTableImport
public Constant<Float> constant(float[][][][] data)
Constant
operationdata
- An array containing the values to put into the new constant. The dimensions of theConstant
public <T extends Number,U extends Number> ScatterMax<T> scatterMax(Operand<T> ref, Operand<U> indices, Operand<T> updates, ScatterMax.Options... options)
ScatterMax
operationref
- Should be from a `Variable` node.indices
- A tensor of indices into the first dimension of `ref`.updates
- A tensor of updated values to reduce into `ref`.options
- carries optional attributes valuesScatterMax
public <T> Merge<T> merge(Iterable<Operand<T>> inputs)
Merge
operationinputs
- The input tensors, exactly one of which will become available.Merge
public MapClear mapClear(List<Class<?>> dtypes, MapClear.Options... options)
MapClear
operationdtypes
- options
- carries optional attributes valuesMapClear
public <T,U extends Number> Lu<T,U> lu(Operand<T> input, Class<U> outputIdxType)
Lu
operationinput
- A tensor of shape `[..., M, M]` whose inner-most 2 dimensions form matrices ofoutputIdxType
- Lu
public ConsumeMutexLock consumeMutexLock(Operand<?> mutexLock)
ConsumeMutexLock
operationmutexLock
- A tensor returned by `MutexLock`.ConsumeMutexLock
public <T,U extends Number,V extends Number> BatchToSpaceNd<T> batchToSpaceNd(Operand<T> input, Operand<U> blockShape, Operand<V> crops)
BatchToSpaceNd
operationinput
- N-D with shape `input_shape = [batch] + spatial_shape + remaining_shape`,blockShape
- 1-D with shape `[M]`, all values must be >= 1.crops
- 2-D with shape `[M, 2]`, all values must be >= 0.BatchToSpaceNd
public <T> UnbatchGrad<T> unbatchGrad(Operand<T> originalInput, Operand<Long> batchIndex, Operand<T> grad, Operand<Long> id, UnbatchGrad.Options... options)
UnbatchGrad
operationoriginalInput
- batchIndex
- grad
- id
- options
- carries optional attributes valuesUnbatchGrad
public <T,U extends Number> Max<T> max(Operand<T> input, Operand<U> axis, Max.Options... options)
Max
operationinput
- The tensor to reduce.axis
- The dimensions to reduce. Must be in the rangeoptions
- carries optional attributes valuesMax
public <T extends Number> NextAfter<T> nextAfter(Operand<T> x1, Operand<T> x2)
NextAfter
operationx1
- x2
- NextAfter
public <T> TensorArrayUnpack tensorArrayUnpack(Operand<String> handle, Operand<T> value, Operand<Float> flowIn)
TensorArrayUnpack
operationhandle
- value
- flowIn
- TensorArrayUnpack
public <U,T> Bitcast<U> bitcast(Operand<T> input, Class<U> type)
Bitcast
operationinput
- type
- Bitcast
public <T,U extends Number,V extends Number> Gather<T> gather(Operand<T> params, Operand<U> indices, Operand<V> axis, Gather.Options... options)
Gather
operationparams
- The tensor from which to gather values. Must be at least rankindices
- Index tensor. Must be in range `[0, params.shape[axis])`.axis
- The axis in `params` to gather `indices` from. Defaults to the firstoptions
- carries optional attributes valuesGather
public Constant<Integer> constant(int data)
Constant
operationdata
- The value to put into the new constant.Constant
public OrderedMapStage orderedMapStage(Operand<Long> key, Operand<Integer> indices, Iterable<Operand<?>> values, List<Class<?>> dtypes, OrderedMapStage.Options... options)
OrderedMapStage
operationkey
- int64indices
- values
- a list of tensorsdtypes
- options
- carries optional attributes valuesOrderedMapStage
public OrderedMapSize orderedMapSize(List<Class<?>> dtypes, OrderedMapSize.Options... options)
OrderedMapSize
operationdtypes
- options
- carries optional attributes valuesOrderedMapSize
public MapUnstageNoKey mapUnstageNoKey(Operand<Integer> indices, List<Class<?>> dtypes, MapUnstageNoKey.Options... options)
MapUnstageNoKey
operationindices
- dtypes
- options
- carries optional attributes valuesMapUnstageNoKey
public LookupTableSize lookupTableSize(Operand<?> tableHandle)
LookupTableSize
operationtableHandle
- Handle to the table.LookupTableSize
public Constant<String> constant(byte[][][][] data)
Constant
operationdata
- An array containing the values to put into the new constant. String elements areConstant
public <T> QuantizedConcat<T> quantizedConcat(Operand<Integer> concatDim, Iterable<Operand<T>> values, Iterable<Operand<Float>> inputMins, Iterable<Operand<Float>> inputMaxes)
QuantizedConcat
operationconcatDim
- 0-D. The dimension along which to concatenate. Must be in thevalues
- The `N` Tensors to concatenate. Their ranks and types must match,inputMins
- The minimum scalar values for each of the input tensors.inputMaxes
- The maximum scalar values for each of the input tensors.QuantizedConcat
public <T> RefSelect<T> refSelect(Operand<Integer> index, Iterable<Operand<T>> inputs)
RefSelect
operationindex
- A scalar that determines the input that gets selected.inputs
- A list of ref tensors, one of which will be forwarded to `output`.RefSelect
public <T,U extends Number> QuantizedReshape<T> quantizedReshape(Operand<T> tensor, Operand<U> shape, Operand<Float> inputMin, Operand<Float> inputMax)
QuantizedReshape
operationtensor
- shape
- Defines the shape of the output tensor.inputMin
- The minimum value of the input.inputMax
- The maximum value of the input.QuantizedReshape
public <T> BatchMatMulV2<T> batchMatMulV2(Operand<T> x, Operand<T> y, BatchMatMulV2.Options... options)
BatchMatMulV2
operationx
- 2-D or higher with shape `[..., r_x, c_x]`.y
- 2-D or higher with shape `[..., r_y, c_y]`.options
- carries optional attributes valuesBatchMatMulV2
public <T> ImmutableConst<T> immutableConst(Class<T> dtype, Shape shape, String memoryRegionName)
ImmutableConst
operationdtype
- Type of the returned tensor.shape
- Shape of the returned tensor.memoryRegionName
- Name of readonly memory region used by the tensor, seeImmutableConst
public <T,U extends Number> ScatterUpdate<T> scatterUpdate(Operand<T> ref, Operand<U> indices, Operand<T> updates, ScatterUpdate.Options... options)
ScatterUpdate
operationref
- Should be from a `Variable` node.indices
- A tensor of indices into the first dimension of `ref`.updates
- A tensor of updated values to store in `ref`.options
- carries optional attributes valuesScatterUpdate
public <T extends Number> VariableShape<T> variableShape(Operand<?> input, Class<T> outType)
VariableShape
operationinput
- outType
- VariableShape
public <T,V extends Number,U extends Number> Unique<T,V> unique(Operand<T> x, Operand<U> axis, Class<V> outIdx)
Unique
operationx
- A `Tensor`.axis
- A `Tensor` of type `int32` (default: None). The axis of the Tensor tooutIdx
- Unique
public <T> RefSwitch<T> refSwitch(Operand<T> data, Operand<Boolean> pred)
RefSwitch
operationdata
- The ref tensor to be forwarded to the appropriate output.pred
- A scalar that specifies which output port will receive data.RefSwitch
public Abort abort(Abort.Options... options)
Abort
operationoptions
- carries optional attributes valuesAbort
public <T> TensorListGetItem<T> tensorListGetItem(Operand<?> inputHandle, Operand<Integer> index, Operand<Integer> elementShape, Class<T> elementDtype)
TensorListGetItem
operationinputHandle
- index
- elementShape
- elementDtype
- TensorListGetItem
public <T extends Number,U extends Number> StatefulRandomBinomial<Long> statefulRandomBinomial(Operand<?> resource, Operand<Long> algorithm, Operand<T> shape, Operand<U> counts, Operand<U> probs)
StatefulRandomBinomial
operationresource
- algorithm
- shape
- counts
- probs
- StatefulRandomBinomial
public <T extends Number> DecodePaddedRaw<T> decodePaddedRaw(Operand<String> inputBytes, Operand<Integer> fixedLength, Class<T> outType, DecodePaddedRaw.Options... options)
DecodePaddedRaw
operationinputBytes
- Tensor of string to be decoded.fixedLength
- Length in bytes for each element of the decoded output. Must be a multipleoutType
- options
- carries optional attributes valuesDecodePaddedRaw
public <T,U extends Number> ReduceProd<T> reduceProd(Operand<T> input, Operand<U> axis, ReduceProd.Options... options)
ReduceProd
operationinput
- The tensor to reduce.axis
- The dimensions to reduce. Must be in the rangeoptions
- carries optional attributes valuesReduceProd
public <U,T> StatefulStandardNormalV2<U> statefulStandardNormalV2(Operand<?> resource, Operand<Long> algorithm, Operand<T> shape, Class<U> dtype)
StatefulStandardNormalV2
operationresource
- The handle of the resource variable that stores the state of the RNG.algorithm
- The RNG algorithm.shape
- The shape of the output tensor.dtype
- The type of the output.StatefulStandardNormalV2
public <T> GuaranteeConst<T> guaranteeConst(Operand<T> input)
GuaranteeConst
operationinput
- GuaranteeConst
public Print print(Operand<String> input, Print.Options... options)
Print
operationinput
- The string scalar to print.options
- carries optional attributes valuesPrint
public <T,U> LookupTableExport<T,U> lookupTableExport(Operand<?> tableHandle, Class<T> Tkeys, Class<U> Tvalues)
LookupTableExport
operationtableHandle
- Handle to the table.Tkeys
- Tvalues
- LookupTableExport
public Constant<String> constant(String data, Charset charset)
Constant
operationcharset
- The encoding from String to bytes.data
- The string to put into the new constant.Constant
public <T> ParallelDynamicStitch<T> parallelDynamicStitch(Iterable<Operand<Integer>> indices, Iterable<Operand<T>> data)
ParallelDynamicStitch
operationindices
- data
- ParallelDynamicStitch
public Constant<Integer> constant(int[][] data)
Constant
operationdata
- An array containing the values to put into the new constant. The dimensions of theConstant
public Constant<Double> constant(double[][][][][] data)
Constant
operationdata
- An array containing the values to put into the new constant. The dimensions of theConstant
public StageSize stageSize(List<Class<?>> dtypes, StageSize.Options... options)
StageSize
operationdtypes
- options
- carries optional attributes valuesStageSize
public OrderedMapClear orderedMapClear(List<Class<?>> dtypes, OrderedMapClear.Options... options)
OrderedMapClear
operationdtypes
- options
- carries optional attributes valuesOrderedMapClear
public <U extends Number,T> ShapeN<U> shapeN(Iterable<Operand<T>> input, Class<U> outType)
ShapeN
operationinput
- outType
- ShapeN
public <T> MulNoNan<T> mulNoNan(Operand<T> x, Operand<T> y)
MulNoNan
operationx
- y
- MulNoNan
public DestroyResourceOp destroyResourceOp(Operand<?> resource, DestroyResourceOp.Options... options)
DestroyResourceOp
operationresource
- handle to the resource to delete.options
- carries optional attributes valuesDestroyResourceOp
public Constant<Double> constant(double data)
Constant
operationdata
- The value to put into the new constant.Constant
public <T extends Number> ReduceAny reduceAny(Operand<Boolean> input, Operand<T> axis, ReduceAny.Options... options)
ReduceAny
operationinput
- The tensor to reduce.axis
- The dimensions to reduce. Must be in the rangeoptions
- carries optional attributes valuesReduceAny
public BarrierTakeMany barrierTakeMany(Operand<String> handle, Operand<Integer> numElements, List<Class<?>> componentTypes, BarrierTakeMany.Options... options)
BarrierTakeMany
operationhandle
- The handle to a barrier.numElements
- A single-element tensor containing the number of elements tocomponentTypes
- The type of each component in a value.options
- carries optional attributes valuesBarrierTakeMany
public <T extends Number,U> ResourceScatterNdAdd resourceScatterNdAdd(Operand<?> ref, Operand<T> indices, Operand<U> updates, ResourceScatterNdAdd.Options... options)
ResourceScatterNdAdd
operationref
- A resource handle. Must be from a VarHandleOp.indices
- A Tensor. Must be one of the following types: int32, int64.updates
- A Tensor. Must have the same type as ref. A tensor ofoptions
- carries optional attributes valuesResourceScatterNdAdd
public <T,U> MutableHashTableOfTensors mutableHashTableOfTensors(Class<T> keyDtype, Class<U> valueDtype, MutableHashTableOfTensors.Options... options)
MutableHashTableOfTensors
operationkeyDtype
- Type of the table keys.valueDtype
- Type of the table values.options
- carries optional attributes valuesMutableHashTableOfTensors
public Barrier barrier(List<Class<?>> componentTypes, Barrier.Options... options)
Barrier
operationcomponentTypes
- The type of each component in a value.options
- carries optional attributes valuesBarrier
public Constant<Long> constant(long[] shape, LongBuffer data)
Constant
operationshape
- the tensor shape.data
- a buffer containing the tensor data.IllegalArgumentException
- If the tensor shape is not compatible with the bufferConstant
public <T> SetDiff1d<T,Integer> setDiff1d(Operand<T> x, Operand<T> y)
SetDiff1d
operationx
- 1-D. Values to keep.y
- 1-D. Values to remove.SetDiff1d
public <T extends Number> ScaleAndTranslate scaleAndTranslate(Operand<T> images, Operand<Integer> size, Operand<Float> scale, Operand<Float> translation, ScaleAndTranslate.Options... options)
ScaleAndTranslate
operationimages
- size
- scale
- translation
- options
- carries optional attributes valuesScaleAndTranslate
public <T extends Number> TensorListElementShape<T> tensorListElementShape(Operand<?> inputHandle, Class<T> shapeType)
TensorListElementShape
operationinputHandle
- shapeType
- TensorListElementShape
public <T> EnsureShape<T> ensureShape(Operand<T> input, Shape shape)
EnsureShape
operationinput
- A tensor, whose shape is to be validated.shape
- The expected (possibly partially specified) shape of the input tensor.EnsureShape
public Constant<Long> constant(long data)
Constant
operationdata
- The value to put into the new constant.Constant
public <T> Where where(Operand<T> condition)
Where
operationcondition
- Where
public MapPeek mapPeek(Operand<Long> key, Operand<Integer> indices, List<Class<?>> dtypes, MapPeek.Options... options)
MapPeek
operationkey
- indices
- dtypes
- options
- carries optional attributes valuesMapPeek
public Constant<Float> constant(float[][][][][] data)
Constant
operationdata
- An array containing the values to put into the new constant. The dimensions of theConstant
public <T extends Number> UnravelIndex<T> unravelIndex(Operand<T> indices, Operand<T> dims)
UnravelIndex
operationindices
- An 0-D or 1-D `int` Tensor whose elements are indices into thedims
- An 1-D `int` Tensor. The shape of the array to use for unravelingUnravelIndex
public <U,T extends Number> ScatterNd<U> scatterNd(Operand<T> indices, Operand<U> updates, Operand<T> shape)
ScatterNd
operationindices
- Index tensor.updates
- Updates to scatter into output.shape
- 1-D. The shape of the resulting tensor.ScatterNd
public <T> AssignSubVariableOp assignSubVariableOp(Operand<?> resource, Operand<T> value)
AssignSubVariableOp
operationresource
- handle to the resource in which to store the variable.value
- the value by which the variable will be incremented.AssignSubVariableOp
public <T> StatefulStandardNormalV2<Float> statefulStandardNormalV2(Operand<?> resource, Operand<Long> algorithm, Operand<T> shape)
StatefulStandardNormalV2
operationresource
- The handle of the resource variable that stores the state of the RNG.algorithm
- The RNG algorithm.shape
- The shape of the output tensor.StatefulStandardNormalV2
public <T,U extends Number> BroadcastTo<T> broadcastTo(Operand<T> input, Operand<U> shape)
BroadcastTo
operationinput
- A Tensor to broadcast.shape
- An 1-D `int` Tensor. The shape of the desired output.BroadcastTo
public <T> Shape<Integer> shape(Operand<T> input)
Shape
operationinput
- Shape
public <T> SwitchCond<T> switchCond(Operand<T> data, Operand<Boolean> pred)
SwitchCond
operationdata
- The tensor to be forwarded to the appropriate output.pred
- A scalar that specifies which output port will receive data.SwitchCond
public BarrierReadySize barrierReadySize(Operand<String> handle)
BarrierReadySize
operationhandle
- The handle to a barrier.BarrierReadySize
public <T,U extends Number> ScatterNdUpdate<T> scatterNdUpdate(Operand<T> ref, Operand<U> indices, Operand<T> updates, ScatterNdUpdate.Options... options)
ScatterNdUpdate
operationref
- A mutable Tensor. Should be from a Variable node.indices
- A Tensor. Must be one of the following types: int32, int64.updates
- A Tensor. Must have the same type as ref. A tensor of updatedoptions
- carries optional attributes valuesScatterNdUpdate
public Constant<Boolean> constant(boolean[][][] data)
Constant
operationdata
- An array containing the values to put into the new constant. The dimensions of theConstant
public StageClear stageClear(List<Class<?>> dtypes, StageClear.Options... options)
StageClear
operationdtypes
- options
- carries optional attributes valuesStageClear
public AssertThat assertThat(Operand<Boolean> condition, Iterable<Operand<?>> data, AssertThat.Options... options)
AssertThat
operationcondition
- The condition to evaluate.data
- The tensors to print out when condition is false.options
- carries optional attributes valuesAssertThat
public TryRpc tryRpc(Operand<String> address, Operand<String> method, Operand<String> request, TryRpc.Options... options)
TryRpc
operationaddress
- `0-D` or `1-D`. The address (i.e. host_name:port) of the RPC server.method
- `0-D` or `1-D`. The method address on the RPC server.request
- `0-D` or `1-D`. Serialized proto strings: the rpc request argument.options
- carries optional attributes valuesTryRpc
public <T> InplaceUpdate<T> inplaceUpdate(Operand<T> x, Operand<Integer> i, Operand<T> v)
InplaceUpdate
operationx
- A tensor of type `T`.i
- A vector. Indices into the left-most dimension of `x`.v
- A `Tensor` of type T. Same dimension sizes as x except the first dimension, which must be the same as i's size.InplaceUpdate
public <T> AssignVariableOp assignVariableOp(Operand<?> resource, Operand<T> value)
AssignVariableOp
operationresource
- handle to the resource in which to store the variable.value
- the value to set the new tensor to use.AssignVariableOp
public Constant<Double> constant(double[][][][] data)
Constant
operationdata
- An array containing the values to put into the new constant. The dimensions of theConstant
public <T> DeepCopy<T> deepCopy(Operand<T> x)
DeepCopy
operationx
- The source tensor of type `T`.DeepCopy
public TensorArrayGrad tensorArrayGrad(Operand<?> handle, Operand<Float> flowIn, String source)
TensorArrayGrad
operationhandle
- The handle to the forward TensorArray.flowIn
- A float scalar that enforces proper chaining of operations.source
- The gradient source string, used to decide which gradient TensorArrayTensorArrayGrad
public <T,U extends Number,V extends Number> Roll<T> roll(Operand<T> input, Operand<U> shift, Operand<V> axis)
Roll
operationinput
- shift
- Dimension must be 0-D or 1-D. `shift[i]` specifies the number of places by whichaxis
- Dimension must be 0-D or 1-D. `axis[i]` specifies the dimension that the shiftRoll
public StagePeek stagePeek(Operand<Integer> index, List<Class<?>> dtypes, StagePeek.Options... options)
StagePeek
operationindex
- dtypes
- options
- carries optional attributes valuesStagePeek
public <T> AssignAddVariableOp assignAddVariableOp(Operand<?> resource, Operand<T> value)
AssignAddVariableOp
operationresource
- handle to the resource in which to store the variable.value
- the value by which the variable will be incremented.AssignAddVariableOp
public <T,U extends Number> Tile<T> tile(Operand<T> input, Operand<U> multiples)
Tile
operationinput
- 1-D or higher.multiples
- 1-D. Length must be the same as the number of dimensions in `input`Tile
public <T> SelectV2<T> selectV2(Operand<Boolean> condition, Operand<T> t, Operand<T> e)
SelectV2
operationcondition
- t
- e
- SelectV2
public <T extends Number> CountUpTo<T> countUpTo(Operand<T> ref, Long limit)
CountUpTo
operationref
- Should be from a scalar `Variable` node.limit
- If incrementing ref would bring it above limit, instead generates anCountUpTo
public Constant<String> constant(byte[][] data)
Constant
operationdata
- An array containing the values to put into the new constant. String elements areConstant
public <T> ReadVariableOp<T> readVariableOp(Operand<?> resource, Class<T> dtype)
ReadVariableOp
operationresource
- handle to the resource in which to store the variable.dtype
- the dtype of the value.ReadVariableOp
public <T> ResourceApplyKerasMomentum resourceApplyKerasMomentum(Operand<?> var, Operand<?> accum, Operand<T> lr, Operand<T> grad, Operand<T> momentum, ResourceApplyKerasMomentum.Options... options)
ResourceApplyKerasMomentum
operationvar
- Should be from a Variable().accum
- Should be from a Variable().lr
- Scaling factor. Must be a scalar.grad
- The gradient.momentum
- Momentum. Must be a scalar.options
- carries optional attributes valuesResourceApplyKerasMomentum
public Constant<Float> constant(float[][][] data)
Constant
operationdata
- An array containing the values to put into the new constant. The dimensions of theConstant
public Constant<Long> constant(long[][] data)
Constant
operationdata
- An array containing the values to put into the new constant. The dimensions of theConstant
public VariableShape<Integer> variableShape(Operand<?> input)
VariableShape
operationinput
- VariableShape
public <T extends Number> ExtractVolumePatches<T> extractVolumePatches(Operand<T> input, List<Long> ksizes, List<Long> strides, String padding)
ExtractVolumePatches
operationinput
- 5-D Tensor with shape `[batch, in_planes, in_rows, in_cols, depth]`.ksizes
- The size of the sliding window for each dimension of `input`.strides
- 1-D of length 5. How far the centers of two consecutive patches are inpadding
- The type of padding algorithm to use.ExtractVolumePatches
public MapStage mapStage(Operand<Long> key, Operand<Integer> indices, Iterable<Operand<?>> values, List<Class<?>> dtypes, MapStage.Options... options)
MapStage
operationkey
- int64indices
- values
- a list of tensorsdtypes
- options
- carries optional attributes valuesMapStage
public OrderedMapUnstageNoKey orderedMapUnstageNoKey(Operand<Integer> indices, List<Class<?>> dtypes, OrderedMapUnstageNoKey.Options... options)
OrderedMapUnstageNoKey
operationindices
- dtypes
- options
- carries optional attributes valuesOrderedMapUnstageNoKey
public <T,U extends Number,V extends Number> SpaceToBatchNd<T> spaceToBatchNd(Operand<T> input, Operand<U> blockShape, Operand<V> paddings)
SpaceToBatchNd
operationinput
- N-D with shape `input_shape = [batch] + spatial_shape + remaining_shape`,blockShape
- 1-D with shape `[M]`, all values must be >= 1.paddings
- 2-D with shape `[M, 2]`, all values must be >= 0.SpaceToBatchNd
public <T extends Number> ResourceCountUpTo<T> resourceCountUpTo(Operand<?> resource, Long limit, Class<T> T)
ResourceCountUpTo
operationresource
- Should be from a scalar `Variable` node.limit
- If incrementing ref would bring it above limit, instead generates anT
- ResourceCountUpTo
public <T> TensorArrayConcat<T> tensorArrayConcat(Operand<?> handle, Operand<Float> flowIn, Class<T> dtype, TensorArrayConcat.Options... options)
TensorArrayConcat
operationhandle
- The handle to a TensorArray.flowIn
- A float scalar that enforces proper chaining of operations.dtype
- The type of the elem that is returned.options
- carries optional attributes valuesTensorArrayConcat
public <T> ShapeN<Integer> shapeN(Iterable<Operand<T>> input)
ShapeN
operationinput
- ShapeN
public <T extends Number,U> EmptyTensorList emptyTensorList(Operand<T> elementShape, Operand<Integer> maxNumElements, Class<U> elementDtype)
EmptyTensorList
operationelementShape
- maxNumElements
- elementDtype
- EmptyTensorList
public <T extends Number> Any any(Operand<Boolean> input, Operand<T> axis, Any.Options... options)
Any
operationinput
- The tensor to reduce.axis
- The dimensions to reduce. Must be in the rangeoptions
- carries optional attributes valuesAny
public <T> TensorArray tensorArray(Operand<Integer> size, Class<T> dtype, TensorArray.Options... options)
TensorArray
operationsize
- The size of the array.dtype
- The type of the elements on the tensor_array.options
- carries optional attributes valuesTensorArray
public <T,U extends Number> StridedSlice<T> stridedSlice(Operand<T> input, Operand<U> begin, Operand<U> end, Operand<U> strides, StridedSlice.Options... options)
StridedSlice
operationinput
- begin
- `begin[k]` specifies the offset into the `k`th range specification.end
- `end[i]` is like `begin` with the exception that `end_mask` isstrides
- `strides[i]` specifies the increment in the `i`th specificationoptions
- carries optional attributes valuesStridedSlice
public <T> Rank rank(Operand<T> input)
Rank
operationinput
- Rank
public Constant<Integer> constant(int[][][][][] data)
Constant
operationdata
- An array containing the values to put into the new constant. The dimensions of theConstant
public <T,U extends Number> ExpandDims<T> expandDims(Operand<T> input, Operand<U> axis)
ExpandDims
operationinput
- axis
- 0-D (scalar). Specifies the dimension index at which toExpandDims
public VarIsInitializedOp varIsInitializedOp(Operand<?> resource)
VarIsInitializedOp
operationresource
- the input resource handle.VarIsInitializedOp
public Constant<Boolean> constant(boolean data)
Constant
operationdata
- The value to put into the new constant.Constant
public <T,U extends Number> Zeros<T> zeros(Operand<U> dims, Class<T> type)
Zeros
operationdims
- a 1-D operand that represents the shape of the output tensortype
- the output tensor datatypeIllegalArgumentException
- if the tensor type or shape cannot be initialized with zeros.Zeros
public <T,U extends Number> TensorScatterSub<T> tensorScatterSub(Operand<T> tensor, Operand<U> indices, Operand<T> updates)
TensorScatterSub
operationtensor
- Tensor to copy/update.indices
- Index tensor.updates
- Updates to scatter into output.TensorScatterSub
public <T> Snapshot<T> snapshot(Operand<T> input)
Snapshot
operationinput
- Snapshot
public <T,U extends Number> Pad<T> pad(Operand<T> input, Operand<U> paddings, Operand<T> constantValues)
Pad
operationinput
- paddings
- constantValues
- Pad
public <T,U extends Number> Concat<T> concat(Iterable<Operand<T>> values, Operand<U> axis)
Concat
operationvalues
- List of `N` Tensors to concatenate. Their ranks and types must match,axis
- 0-D. The dimension along which to concatenate. Must be in theConcat
public BarrierClose barrierClose(Operand<String> handle, BarrierClose.Options... options)
BarrierClose
operationhandle
- The handle to a barrier.options
- carries optional attributes valuesBarrierClose
public <T> Squeeze<T> squeeze(Operand<T> input, Squeeze.Options... options)
Squeeze
operationinput
- The `input` to squeeze.options
- carries optional attributes valuesSqueeze
public <T> Stack<T> stack(Iterable<Operand<T>> values, Stack.Options... options)
Stack
operationvalues
- Must be of same shape and type.options
- carries optional attributes valuesStack
public <T> NextIteration<T> nextIteration(Operand<T> data)
NextIteration
operationdata
- The tensor to be made available to the next iteration.NextIteration
public TensorArraySize tensorArraySize(Operand<?> handle, Operand<Float> flowIn)
TensorArraySize
operationhandle
- The handle to a TensorArray (output of TensorArray or TensorArrayGrad).flowIn
- A float scalar that enforces proper chaining of operations.TensorArraySize
public OrderedMapUnstage orderedMapUnstage(Operand<Long> key, Operand<Integer> indices, List<Class<?>> dtypes, OrderedMapUnstage.Options... options)
OrderedMapUnstage
operationkey
- indices
- dtypes
- options
- carries optional attributes valuesOrderedMapUnstage
public <T> GetSessionHandle getSessionHandle(Operand<T> value)
GetSessionHandle
operationvalue
- The tensor to be stored.GetSessionHandle
public <T,U extends Number> GatherNd<T> gatherNd(Operand<T> params, Operand<U> indices)
GatherNd
operationparams
- The tensor from which to gather values.indices
- Index tensor.GatherNd
public <T,U extends Number> ScatterNdAdd<T> scatterNdAdd(Operand<T> ref, Operand<U> indices, Operand<T> updates, ScatterNdAdd.Options... options)
ScatterNdAdd
operationref
- A mutable Tensor. Should be from a Variable node.indices
- A Tensor. Must be one of the following types: int32, int64.updates
- A Tensor. Must have the same type as ref. A tensor of updated valuesoptions
- carries optional attributes valuesScatterNdAdd
public <T> Variable<T> variable(Shape shape, Class<T> dtype, Variable.Options... options)
Variable
operationshape
- The shape of the variable tensor.dtype
- The type of elements in the variable tensor.options
- carries optional attributes valuesVariable
public Constant<Long> constant(long[][][][][][] data)
Constant
operationdata
- An array containing the values to put into the new constant. The dimensions of theConstant
public <T> Split<T> split(Operand<Integer> axis, Operand<T> value, Long numSplit)
Split
operationaxis
- 0-D. The dimension along which to split. Must be in the rangevalue
- The tensor to split.numSplit
- The number of ways to split. Must evenly divideSplit
public StringLower stringLower(Operand<String> input, StringLower.Options... options)
StringLower
operationinput
- options
- carries optional attributes valuesStringLower
public <T extends Number> NonMaxSuppressionV5<T> nonMaxSuppressionV5(Operand<T> boxes, Operand<T> scores, Operand<Integer> maxOutputSize, Operand<T> iouThreshold, Operand<T> scoreThreshold, Operand<T> softNmsSigma, NonMaxSuppressionV5.Options... options)
NonMaxSuppressionV5
operationboxes
- A 2-D float tensor of shape `[num_boxes, 4]`.scores
- A 1-D float tensor of shape `[num_boxes]` representing a singlemaxOutputSize
- A scalar integer tensor representing the maximum number ofiouThreshold
- A 0-D float tensor representing the threshold for deciding whetherscoreThreshold
- A 0-D float tensor representing the threshold for deciding when to removesoftNmsSigma
- A 0-D float tensor representing the sigma parameter for Soft NMS; see Bodla etoptions
- carries optional attributes valuesNonMaxSuppressionV5
public <T> Placeholder<T> placeholder(Class<T> dtype, Placeholder.Options... options)
Placeholder
operationdtype
- The type of elements in the tensor.options
- carries optional attributes valuesPlaceholder
public ControlTrigger controlTrigger()
ControlTrigger
operationControlTrigger
public <T,U extends Number> Reshape<T> reshape(Operand<T> tensor, Operand<U> shape)
Reshape
operationtensor
- shape
- Defines the shape of the output tensor.Reshape
public <T> TensorArrayPack<T> tensorArrayPack(Operand<String> handle, Operand<Float> flowIn, Class<T> dtype, TensorArrayPack.Options... options)
TensorArrayPack
operationhandle
- flowIn
- dtype
- options
- carries optional attributes valuesTensorArrayPack
public Ops withSubScope(String childScopeName)
Scope#withSubScope(String)}
public Ops withName(String opName)
Scope#withName(String)}
public Ops withControlDependencies(Iterable<Operand<?>> controls)
Scope#withControlDependencies(Iterable>)}
public final SummaryOps summary()
summary
operationspublic final NnOps nn()
nn
operationspublic final ImageOps image()
image
operationspublic final DataOps data()
data
operationspublic final IoOps io()
io
operationspublic final DtypesOps dtypes()
dtypes
operationspublic final LinalgOps linalg()
linalg
operationspublic final RandomOps random()
random
operationspublic final StringsOps strings()
strings
operationspublic final SparseOps sparse()
sparse
operationspublic final BitwiseOps bitwise()
bitwise
operationspublic final MathOps math()
math
operationspublic final AudioOps audio()
audio
operationspublic final SignalOps signal()
signal
operationspublic final TrainOps train()
train
operationspublic final QuantizationOps quantization()
quantization
operationspublic static Ops create(ExecutionEnvironment env)
public static Ops create()
Invoking this method is equivalent to Ops.create(EagerSession.getDefault())
.
Copyright © 2022. All rights reserved.