".github/vscode:/vscode.git/clone" did not exist on "e4a7e5b2cad59a5a0df6cd6c3d008be7c779dcb0"
-
Michael Yang authored
The current splitDim function only operates on tensors that are split evenly which isn't always the case, e.g. a QKV tensor. This change allows the function to be used for arbitrary splits
45f56355
