np.eye(3)生产一个单位矩阵3*3的单位矩阵
当np.eye(3)后面跟一个整数i时,此时取得是该矩阵的第i行。
tp1=np.eye(3)
tp1=[[1,0,0],
[0,1,0],
[0,0,1]]
tp1[2]取该矩阵的第2行,为 [0,0,1]
当np.eye(3)后面跟一个数列时,此时取得是该矩阵的多个行
tp1[1][2]取的是矩阵索引为1,2的行;分别为
[0,1,0]
[0,0,1]
该结果恰好是对索引的one-hot处理
tp1=([[1, 2, 3],
[4, 5, 6],
[7, 8, 9]])
tp1[[1,2]]表示的是取tp1的第1行和第2行
tp[[1,2],[0,1]]表示的取tp1的[1][0]位置的元素和tp1的[2][1]位置的元素
等同于tp1[1][0] tp1[2][1]