- Type Parameters:
T -
- All Superinterfaces:
- Serializable
- All Known Subinterfaces:
- ModelAdapter<T>
- All Known Implementing Classes:
- ArgmaxAdapter, Regression2dAdapter, YoloModelAdapter
public interface OutputAdapter<T>
extends Serializable
This interface describes entity used to conver neural network output to specified class.
I.e. INDArray -> int[] on the fly.
PLEASE NOTE: Implementation will be used in workspace environment to avoid additional allocations during inference.
This means you shouldn't store or return the INDArrays passed to OutputAdapter.apply(INDArray...) directly.
If you need a copy of the output array, use standard network output methods, or use INDArray.detach() before storing the array