You could use the scikit-learn implementation 3.2k. Just get your predictions and targets using .numpy().
Alternatively you could of course calculate the conf matrix manually, if you don’t want to install/use scikit-learn.