Skip to main content
torch.js has not been released yet.
torch.js logotorch.js logotorch.js
PlaygroundContact
Login
Documentation
IntroductionType SafetyTensor ExpressionsTensor IndexingEinsumEinopsAutogradTraining a ModelProfiling & MemoryPyTorch MigrationBest PracticesRuntimesPerformancePyTorch CompatibilityBenchmarksDType Coverage
IntroductionRenderer GalleryRenderersAnalysis
computeQuantilescomputePercentilescomputeCorrelationcomputeCovariancedetectOutliersdetectOutliersZScorecomputeExtendedStatsdescribeExtendedStatsQuantileResultCorrelationResultOutlierResultExtendedStatscreateDBSCANdbscanDBSCANDBSCAN.fitDBSCAN.destroyDBSCANOptionsDBSCANResultcreateHierarchicalClusteringhierarchicalClusteringHierarchicalClusteringHierarchicalClustering.fitHierarchicalClustering.destroyLinkageMethodDistanceMetricHierarchicalOptionsDendrogramNodeHierarchicalResultcomputeHistogramgetHistogramCentersnormalizeHistogramHistogramResultcreateKDEkdeKDEKDE.fitKDE.destroyKernelTypeKDEOptionsKDEResultcreateKMeanskmeansKMeansKMeans.fitKMeans.destroyKMeansOptionsKMeansResultcreatePCApcaPCAPCA.fitPCA.destroyPCAOptionsPCAResultthrottleProgressconsoleProgressProgressInfoProgressCallbackcomputeStatscomputeMinMaxdescribeStatsTensorStatscreateTSNEtsneTSNETSNE.fitTSNE.destroyTSNEOptionsTSNEResultcreateUMAPumapUMAPUMAP.fitUMAP.destroyUMAPOptionsUMAPResult
torch.js· 2026
LegalTerms of UsePrivacy Policy
/
/
  1. docs
  2. viz
  3. viz
  4. analysis
  5. createKMeans

viz.analysis.createKMeans

function createKMeans(): KMeans

Create a new KMeans clustering instance.

Factory function for creating KMeans instances. Useful when you need to cluster multiple datasets and want to reuse GPU pipelines.

Returns

KMeans– New KMeans instance with uninitialized GPU resources

Examples

// Reuse pipelines for multiple datasets
const km = createKMeans();
const result1 = await km.fit(data1, [n1, d], { k: 5 });
const result2 = await km.fit(data2, [n2, d], { k: 10 });
km.destroy();

// Or use kmeans() convenience function for single clustering
const result = await kmeans(data, [n, d], { k: 5 });

See Also

  • kmeans - For single-use clustering (handles cleanup automatically)
Previous
createKDE
Next
createPCA