生成one-hot矩阵

生成一个大小为 (n, m) 的 one-hot 矩阵可以使用以下代码：

import numpy as np

def one_hot(n, m, index):
    """
    生成一个大小为 (n, m) 的 one-hot 矩阵，第 index 个位置为 1，其余位置为 0。
    """
    hot = np.zeros((n, m))
    hot[index] = 1
    return hot

其中，n 表示矩阵的行数，m 表示矩阵的列数，index 表示需要将矩阵中的哪个位置设置为 1。例如，生成一个大小为 (5, 3) 的 one-hot 矩阵，其中第 2 行为 1，可以使用以下代码：

hot_matrix = one_hot(5, 3, 1)
print(hot_matrix)

输出结果为：

[[0. 0. 0.]
 [1. 0. 0.]
 [0. 0. 0.]
 [0. 0. 0.]
 [0. 0. 0.]]