scSLAT.model.train.feature_reconstruct_loss
- scSLAT.model.train.feature_reconstruct_loss(embd, x, recon_model)[source]
Reconstruction loss (MSE)
- Parameters:
embd (
torch.Tensor) – embd of a cellx (
torch.Tensor) – inputrecon_model (
torch.nn.modules.module.Module) – reconstruction model
- Return type: