numpy的eye用于生成单位矩阵:
import numpy as np # 创建3阶单位矩阵 a_eye = np.eye(3) print(a_eye)
执行结果:
[[1. 0. 0.] [0. 1. 0.] [0. 0. 1.]]
上一篇 numpy
下一篇 torchvision.transforms
版权所有 (c)2021-2022 MSHXW.COM
ICP备案号:晋ICP备2021003244-6号