TensorFlow 2 version | View source on GitHub |
Retrieves the elements of indices indices
in the tensor reference
.
tf.keras.backend.gather(
reference, indices
)
Arguments | |
---|---|
reference
|
A tensor. |
indices
|
An integer tensor of indices. |
Returns | |
---|---|
A tensor of same type as reference .
|