Skip to content

gather

Gathering.

This layer gathers elements by indices along axis.


Parameters

  • axis : axis to be gathered.

Weights

  • indices : gathering indices.