The error message indicates you are attempting to apply a conv operation to a single N0f8 value rather than an array of them. Would you mind showing your loss function implementation? Perhaps there is an extra broadcast operation.
The error message indicates you are attempting to apply a conv operation to a single N0f8 value rather than an array of them. Would you mind showing your loss function implementation? Perhaps there is an extra broadcast operation.