deepfold.utils.rigid_utils.identity_rot_mats¶

deepfold.utils.rigid_utils.identity_rot_mats(batch_dims: Tuple[int], dtype: dtype | None = None, device: device | None = None, requires_grad: bool = True) → Tensor[source]¶