std :: ml::module
#include <std/ml/module.h> source
#include <std/ml/module.h>
#define TRAIN_LOSS_BCE 2
binary cross-entropy on logits (tensor_bce_logits_loss)