GatherDimensionNumbers

Represents the gather dimension numbers.

Usage

GatherDimensionNumbers(
  offset_dims,
  collapsed_slice_dims,
  operand_batching_dims = integer(),
  start_indices_batching_dims = integer(),
  start_index_map,
  index_vector_dim
)

Arguments

offset_dims

(integer())
The offset dimensions.

collapsed_slice_dims

(integer())
The collapsed slice dimensions.

operand_batching_dims

(integer())
The operand batching dimensions.

start_indices_batching_dims

(integer())
The start indices batching dimensions.

start_index_map

(integer())
Maps start indices to operand dimensions.

index_vector_dim

(integer(1))
The index vector dimension.