Tried using single pointer to tensors instead