KNN中,训练样本有train_count个,测试样本有test_count个,每个样本有attr_count个属性.现在需要快速计算test_count个测试样本和train_count个样本之间的距离,距离表示为一个test_count行.train_count列的矩阵. 这个问题可以转化为矩阵乘法,而矩阵乘法可以使用GPU进行加速. import numpy as np train_count = 3 test_count = 2 attr_count = 4 train = np.ran