nevermind. I think it's faulty custom tensor extraction function