You could use the ROC implementations from other libraries such as sklearn.metrics.roc_curve 257. To do so you could transform the predictions and targets to numpy arrays via tensor.numpy() and apply the mentioned method.