o }oi@s\ddlmZmZddlmZmZmZddlmZm Z m Z m Z dgZ Gdddej eeZ dS))Tensornn) SerializationTyping typecheck) LabelsTypeLossType NeuralTypeRegressionValuesTypeMSELosscs`eZdZdZeddZeddZddeffdd Ze d e d e d e ffd d Z Z S)r z MSELoss cCs"ttdtttdtdS)z3Returns definitions of module input ports. B)predslabels)r tupler rselfr[/home/ubuntu/.local/lib/python3.10/site-packages/nemo/collections/common/losses/mse_loss.py input_typesszMSELoss.input_typescCsdttdiS)z4Returns definitions of module output ports. loss) elements_type)r rrrrr output_types%szMSELoss.output_typesmean reductioncstj|ddS)zS Args: reduction: type of the reduction over the batch )rN)super__init__)rr __class__rrr+szMSELoss.__init__r rreturncst||S)zk Args: preds: output of the classifier labels: ground truth labels )rforward)rr rrrrr2szMSELoss.forward)r) __name__ __module__ __qualname____doc__propertyrrstrrrrr __classcell__rrrrr s  $N)torchrrnemo.core.classesrrrnemo.core.neural_typesrrr r __all__r rrrrs