如何使用PyTorch对数组中的数字执行无监督聚类

import torch
from torch import nn
from torch.utils.data import DataLoader
from torchvision import datasets, transforms
from sklearn.cluster import KMeans

# 假设数组为data_array
data_tensor = torch.tensor(data_array)

# 对数据进行归一化处理
data_tensor = (data_tensor - data_tensor.mean()) / data_tensor.std()

# 使用KMeans算法进行聚类
kmeans = KMeans(n_clusters=num_clusters)

kmeans.fit(data_tensor)

cluster_labels = kmeans.labels_

# 使用matplotlib等库进行可视化