Class RandomExperimentalOps

java.lang.Object

org.tensorflow.op.RandomExperimentalOps

public final class RandomExperimentalOps
extends Object

An API for building random.experimental operations as Ops

See Also:

• Ops

Method Summary

Modifier and Type	Method	Description
final Ops	ops()	Get the parent Ops object.
<T extends TType> StatelessShuffle<T>	statelessShuffle(Operand<T> value, Operand<? extends TType> key, Operand<? extends TType> counter, Operand<TInt32> alg)	Randomly and deterministically shuffles a tensor along its first dimension.

Methods inherited from class Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Method Details

statelessShuffle

public <T extends TType> StatelessShuffle<T> statelessShuffle(Operand<T> value,
 Operand<? extends TType> key,
 Operand<? extends TType> counter,
 Operand<TInt32> alg)

Randomly and deterministically shuffles a tensor along its first dimension. The tensor is shuffled along dimension 0, such that each value[j] is mapped to one and only one output[i]. For example, a mapping that might occur for a 3x2 tensor is:

 [[1, 2],       [[5, 6],
  [3, 4],  ==>   [1, 2],
  [5, 6]]        [3, 4]]
 

The outputs are a deterministic function of value, key, counter and alg.

Type Parameters:

T - data type for StatelessShuffle output and operands

Parameters:

value - The tensor to be shuffled.

key - Key for the counter-based RNG algorithm (shape uint64[1]).

counter - Initial counter for the counter-based RNG algorithm (shape uint64[2] or uint64[1] depending on the algorithm). If a larger vector is given, only the needed portion on the left (i.e. [:N]) will be used.

alg - The RNG algorithm (shape int32[]).

Returns:

a new instance of StatelessShuffle

ops

public final Ops ops()

Get the parent Ops object.