Solved by tom in post #2 
              
              
                If you pass in None, it will instantiate the right thing: 


I’m not quite sure whether the authors of the code want to achieve something specific or whether the interface of optionally passing in the tensors is only accidental (the initialization seems very unidiomatic, too) – in theory that should…