System information

• Have I written custom code (as opposed to using a stock example script provided in TensorFlow.js): Yes
• OS Platform and Distribution (e.g., Linux Ubuntu 16.04): macOS 13.5.2 (22G91)
• Mobile device (e.g. iPhone 8, Pixel 2, Samsung Galaxy) if the issue happens on mobile device: N/A
• TensorFlow.js installed from (npm or script link): script link
• TensorFlow.js version (use command below): 4.11.0
• Browser version: Google Chrome Version 116.0.5845.187 (Official Build) (x86_64)
• Tensorflow.js Converter Version: N/A

Describe the current behavior

2 Tensors are leaked after calling tfvis.metrics.confusionMatrix.

Describe the expected behavior

No tensor is leaked after calling tfvis.metrics.confusionMatrix.

Standalone code to reproduce the issue

Open Google Chrome, navigate to about:blank, then paste the following code in the console.

await import('https://cdn.jsdelivr.net/npm/@tensorflow/[email protected]/dist/tf.min.js')
await import('https://cdn.jsdelivr.net/npm/@tensorflow/[email protected]/dist/tfjs-vis.umd.min.js')

console.log('num of tensors before:', tf.memory().numTensors);

const labels = tf.tensor1d([0]);
const predictions = tf.tensor1d([0]);
const confusionMatrix = await tfvis.metrics.confusionMatrix(labels, predictions);

labels.dispose();
predictions.dispose();

console.log('num of tensors after:', tf.memory().numTensors)

Other info / logs

Console logs:

num of tensors before: 0
num of tensors after: 2