Function autograd::ops::gather [−] [src]

pub fn gather(param: &Tensor, indices: &Tensor, axis: isize) -> Tensor

Gather slices.

For example, this can be used for vector lookup. See https://www.tensorflow.org/api_docs/python/tf/gather.

Arguments

  • param - Target of slicing.
  • indices - Index tensor with arbitrary shape.
  • axis - Slices sub tensors along this axis.

Returns

Tensor with shape (param.shape[..axis] + indices.shape + param.shape[axis+1..])