theislab / chemCPA

Code for "Predicting Cellular Responses to Novel Drug Perturbations at a Single-Cell Resolution", NeurIPS 2022.
https://arxiv.org/abs/2204.13545
MIT License
88 stars 23 forks source link

Unsqueeze tensors if batch size is 1 #135

Closed MxMstrmn closed 10 months ago

MxMstrmn commented 10 months ago

Fixes #130