Gather slices with indices=-1
mapped to default
.
text.gather_with_default(
params, indices, default, name=None, axis=0
)
This operation is similar to tf.gather()
, except that any value of -1
in indices
will be mapped to default
. Example:
gather_with_default(['a', 'b', 'c', 'd'], [2, 0, -1, 2, -1], '_')
<tf.Tensor: shape=(5,), dtype=string,
numpy=array([b'c', b'a', b'_', b'c', b'_'], dtype=object)>
Args |
params
|
The Tensor from which to gather values. Must be at least rank
axis + 1 .
|
indices
|
The index Tensor . Must have dtype int32 or int64 , and values
must be in the range [-1, params.shape[axis]) .
|
default
|
The value to use when indices is -1 . default.shape must
be equal to params.shape[axis + 1:] .
|
name
|
A name for the operation (optional).
|
axis
|
The axis in params to gather indices from. Must be a scalar
int32 or int64 . Supports negative indices.
|
Returns |
A Tensor with the same type as param , and with shape
params.shape[:axis] + indices.shape + params.shape[axis + 1:] .
|