\brief Add a reduce layer to the network. \param input The input tensor to the layer. \param operation The reduction operation to perform. \param reduceAxes The reduction dimensions. The bit in position i of bitmask reduceAxes corresponds to explicit dimension i if result. E.g., the least significant bit corresponds to the first explicit dimension and the next to least significant bit corresponds
| 7321 | //! \return The new reduce layer, or nullptr if it could not be created. |
| 7322 | //! |
| 7323 | IReduceLayer* addReduce( |
| 7324 | ITensor& input, ReduceOperation operation, uint32_t reduceAxes, bool keepDimensions) noexcept |
| 7325 | { |
| 7326 | return mImpl->addReduce(input, operation, reduceAxes, keepDimensions); |
| 7327 | } |
| 7328 | |
| 7329 | //! |
| 7330 | //! \brief Add a TopK layer to the network. |