Add contiguous() call just in case

86613a95 · Gustaf Ahdritz · GitHub · 03518fd1 · 86613a95
Unverified Commit 86613a95 authored Jul 29, 2022 by Gustaf Ahdritz Committed by GitHub Jul 29, 2022
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 0 deletions

openfold/utils/rigid_utils.py openfold/utils/rigid_utils.py +1 -0

No files found.
--- a/openfold/utils/rigid_utils.py
+++ b/openfold/utils/rigid_utils.py
@@ -96,6 +96,7 @@ def identity_rot_mats(
    )
    rots = rots.view(*((1,) * len(batch_dims)), 3, 3)
    rots = rots.expand(*batch_dims, -1, -1)
+    rots = rots.contiguous()

    return rots