理解：2个样本数据，每个数据3行(T_x = 3)，4列(n_value = 4)
indices = np.argmax(pred, 2)
例如，对第一个样本数据的，第一行中的，所有数据取最大值的索引下标
