torch_linear_assignment
