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
104 stars 24 forks source link

Unsqueeze tensors if batch size is 1 #135

Closed MxMstrmn closed 1 year ago

MxMstrmn commented 1 year ago

Fixes #130