std :: ml::module
#include <std/ml/module.h> source
#include <std/ml/module.h>
#define TRAIN_LOSS_MSE 1
mean squared error (tensor_mse_loss)