torch.linalg.lstsq: (Batch element 0): The least squares solution could not be computed because the input matrix does not have full rank (error code: 13).

[zefengzhang123]