\brief Add gather with mode GatherMode::kDEFAULT and specified axis and nbElementWiseDims=0. \param data The tensor to gather values from. \param indices The tensor to get indices from to populate the output tensor. \param axis The axis in the data tensor to gather on. \see IGatherLayer \return The new gather layer, or nullptr if it could not be created.
| 7372 | //! \return The new gather layer, or nullptr if it could not be created. |
| 7373 | //! |
| 7374 | IGatherLayer* addGather(ITensor& data, ITensor& indices, int32_t axis) noexcept |
| 7375 | { |
| 7376 | return mImpl->addGather(data, indices, axis); |
| 7377 | } |
| 7378 | |
| 7379 | //! |
| 7380 | //! \brief Add gather with specified mode, axis=0 and nbElementWiseDims=0. |