本文详细介绍了numpy库中的numpy.eye()和np.identity()函数，重点解析了它们的参数和作用。numpy.eye()不仅用于创建对角线元素为1的矩阵，还可将标签数组转换为one-hot编码。而np.identity()则专门生成主对角线为1的方阵。通过示例代码展示了如何使用这两个函数，并解释了它们在深度学习中将数组转化为one-hot形式的应用。
numpy.eye(N,M=None, k=0, dtype=<type ‘float’>)
关注第一个第三个参数就行了
第一个参数：输出方阵（行数=列数）的规模，即行数或列数
第三个参数：默认情况下输出的是对角线全“1”，其余全“0”的方阵，如果k为正整数，则在右上方第k条对角线全“1”其余全“0”，k为负整数则在左下方第k条对角线全“1”其余全“0”。
原文链接：https://blog.csdn.net/chixujohnny/article/details/51011931
>>> np.eye(2, dtype=int)
array([[1, 
最低0.47元/天 解锁文章
扫一扫
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
