DL4J实战之五：矩阵操作基本功

• 执行结果

全一矩阵

维度 : 2

形状 : [2, 3]

完整矩阵 :

[[ 1.0000, 1.0000, 1.0000],

[ 1.0000, 1.0000, 1.0000]]

3. 全是指定值的矩阵：valueArrayOf

// 创建2行3列的全是指定值的矩阵

INDArray indArray2 = Nd4j.valueArrayOf(new int[] {2, 3}, 888);

disp(“全是指定值(888)的矩阵”, indArray2);

• 执行结果

全是指定值(888)的矩阵

维度 : 2

形状 : [2, 3]

完整矩阵 :

[[ 888.0000, 888.0000, 888.0000],

[ 888.0000, 888.0000, 888.0000]]

4. rand：随机矩阵（0到1之间的随机数）

// 创建2行3列的随机矩阵

INDArray indArray2 = Nd4j.rand(2, 3);

disp(“随机矩阵”, indArray2);

• 执行结果

随机矩阵

维度 : 2

形状 : [2, 3]

完整矩阵 :

[[ 0.7236, 0.5159, 0.1908],

[ 0.9458, 0.4413, 0.4173]]

5. 随机高斯分布的矩阵(平均值为0,标准差为1)：randn

// 创建2行3列的随机高斯分布矩阵

INDArray indArray3 = Nd4j.randn(2, 3);

disp(“随机高斯分布矩阵”, indArray3);

• 执行结果

随机高斯分布矩阵

维度 : 2

形状 : [2, 3]

完整矩阵 :

[[ -0.4287, -0.5790, 0.5004],

[ -0.5122, 1.0551, -0.1998]]

6. 等差数列：linspace

// 创建等差数列，

// 从1到6、长度为10的等差数列

INDArray indArray4 = Nd4j.linspace(1,6, 10);

disp(“等差数列”, indArray4);

• 执行结果

等差数列

维度 : 1

形状 : [10]

完整矩阵 :

[ 1.0000, 1.5556, 2.1111, 2.6667, 3.2222, 3.7778, 4.3333, 4.8889, 5.4444, 6.0000]

7. 根据数组创建矩阵：create(float[] data, int[] shape)

// 根据数组创建2行3列的矩阵

INDArray indArray6 = Nd4j.create(new float[] {1, 2, 3, 4, 5, 6}, new int[] {2,3});

disp(“根据数组创建矩阵”, indArray6);

• 执行结果

根据数组创建矩阵

维度 : 2

形状 : [2, 3]

完整矩阵 :

[[ 1.0000, 2.0000, 3.0000],

[ 4.0000, 5.0000, 6.0000]]

8. 三维矩阵

// 三维矩阵

INDArray indArray7 = Nd4j.valueArrayOf(new int[] {2, 2, 3}, 888);

disp(“三维矩阵”, indArray7);

三维矩阵

维度 : 3

形状 : [2, 2, 3]

完整矩阵 :

[[[ 888.0000, 888.0000, 888.0000],

[ 888.0000, 888.0000, 888.0000]],

[[ 888.0000, 888.0000, 888.0000],

[ 888.0000, 888.0000, 888.0000]]]

9. 创建正方形二维矩阵，并且对角线上的元素值都是1.0：

// 创建3行3列的二维矩阵，对角线值为1.0

INDArray indArray10 = Nd4j.eye(3);

disp(“3*3矩阵，且对角线都是1.0”, indArray10);

• 执行结果

3*3矩阵，且对角线都是1.0

维度 : 2

形状 : [3, 3]

完整矩阵 :

[[ 1.0000, 0,