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`.
Public Methods
Output<T> |
asOutput()
Returns the symbolic handle of a tensor.
|
static <T> StatelessShuffle<T> | |
Output<T> |
output()
A tensor of same shape and type as `value`, shuffled along its first
dimension.
|
Inherited Methods
Public Methods
public Output<T> asOutput ()
Returns the symbolic handle of a tensor.
Inputs to TensorFlow operations are outputs of another TensorFlow operation. This method is used to obtain a symbolic handle that represents the computation of the input.
public static StatelessShuffle<T> create (Scope scope, Operand<T> value, Operand<?> key, Operand<?> counter, Operand<Integer> alg)
Factory method to create a class wrapping a new StatelessShuffle operation.
Parameters
scope | current scope |
---|---|
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
public Output<T> output ()
A tensor of same shape and type as `value`, shuffled along its first dimension.