createUNetModelInput
private fun createUNetModelInput(encoderHiddenStates: OnnxTensor, sample: OnnxTensor, timeStep: OnnxTensor): Map<String, OnnxTensor>(source)
Creates the SDAI value produced by createUNetModelInput.
Return
Result produced by createUNetModelInput.
Author
Dmitriy Moroz
Parameters
encoderHiddenStates
encoder hidden states value consumed by the API.
sample
sample value consumed by the API.
timeStep
time step value consumed by the API.