Closed scott-vsi closed 1 year ago
Looks good. Technically the input could also sometimes be torch tensors when batched = True, but the current implementation actually crashes for that case and the new one would be fine, so just gonna merge it.
Looks good. Technically the input could also sometimes be torch tensors when batched = True, but the current implementation actually crashes for that case and the new one would be fine, so just gonna merge it.