| 400 | |
| 401 | |
| 402 | class World2Local(torch.nn.Module): |
| 403 | def __init__(self): |
| 404 | super().__init__() |
| 405 | |
| 406 | |
| 407 | def forward(self, strands_tbn, strands_positions, root_normals): |
| 408 | return world_to_tbn_space(strands_tbn, strands_positions, root_normals) |
| 409 | |
| 410 | |
| 411 |