大家好,我是奇点。
NDArray
支持大量的运算符(operator)。但因为Java不支持运算符过载, ,所以在DJL中,常见的标准算数运算符(+,—,∗∗,/和∗∗∗∗)都通过函数来实现。可以对任意维度中任何同一维度的张量进行运算。我们可以对任意两个相同维度的张量进行基础运算。
在接下来的例子中,我们用逗号构建两个含有4个元素的元组,来演示一下NDArray
在运算。
简单运算
NDArray x = manager.create(new float[] {
1f, 2f, 4f, 8f });
NDArray y = manager.create(new float[] {
2f, 2f, 2f, 2f });
// x + y
System.out.println("x + y :");
System.out.println(x.add(y));
// x -y
System.out.println("x - y :");
System.out.println(x.sub(y));
// x*y
System.out.println("x * y :"