COSIE.model_component.Prediction_mlp.forward

Prediction_mlp.forward(x)

Perform full forward pass.

Parameters

xtorch.Tensor

Input cell embedding.

Returns

outputtorch.Tensor

Predicted cell embedding.