Why not use the same sklearn analysis to check the principal components? See here

  
      
      Agnès Mustar – 1 Nov 17
  
  
    

Principal Component Analysis (PCA) implemented with PyTorch | Agnès Mustar 66

What is PCA ? PCA is an algorithm capable of finding patterns in data, it is used to reduce the dimension of the data. If X is a matrix of size (m, n). We want to find an encoding fonction f such as f(X) = C where C is a matrix of size (m,l) with …...