Function gather

tensorflow/tensorflow/python/ops/array_ops.py:3867–3956 · view source on GitHub ↗

r"""Gather slices from params axis axis according to indices. Gather slices from params axis `axis` according to `indices`. `indices` must be an integer tensor of any dimension (usually 0-D or 1-D). For 0-D (scalar) `indices`: > `output`$$[p_0, ..., p_{axis-1}, \hspace{5.

(params,
           indices,
           validate_indices=None,
           name=None,
           axis=None,
           batch_dims=0)

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers 4

_apply_mask_1dFunction · 0.70

sparse_maskFunction · 0.70

gather_v2Function · 0.70

_batch_gatherFunction · 0.70

Calls 1

sparse_readMethod · 0.45

Tested by

no test coverage detected