(L)
| 65 | return helper |
| 66 | |
| 67 | def strip_lowerdiag(L): |
| 68 | uncertainty = torch.zeros((L.shape[0], 6), dtype=torch.float, device="cuda") |
| 69 | |
| 70 | uncertainty[:, 0] = L[:, 0, 0] |
| 71 | uncertainty[:, 1] = L[:, 0, 1] |
| 72 | uncertainty[:, 2] = L[:, 0, 2] |
| 73 | uncertainty[:, 3] = L[:, 1, 1] |
| 74 | uncertainty[:, 4] = L[:, 1, 2] |
| 75 | uncertainty[:, 5] = L[:, 2, 2] |
| 76 | return uncertainty |
| 77 | |
| 78 | def strip_symmetric(sym): |
| 79 | return strip_lowerdiag(sym) |