nd4j中的简单卷积

Question

我无法在nd4j中使用简单的卷积，关于这个特定主题的文档很少。我想做的事：

INDArray values = Nd4j.create(new double[]{1, 2, 3, 4, 5, 6, 7, 8, 9, 10});
INDArray kernel = Nd4j.create(new double[]{0.5,0.5});
INDArray conv = Nd4j.getConvolution().convn(values, kernel, Convolution.Type.VALID);

无论值或卷积类型如何，我总是得到相同的异常(见下文)。当nd4j试图将值的数组转换为复数数组以执行我认为的傅立叶变换时，错误似乎会发生。

我试过几个版本的nd4j(0.9.1-0.8.0-0.7.0)，但都没有用。有人能帮忙吗？

Exception in thread "main" java.lang.UnsupportedOperationException
at org.nd4j.linalg.api.complex.BaseComplexNDArray.putScalar(BaseComplexNDArray.java:1947)
at org.nd4j.linalg.api.complex.BaseComplexNDArray.putScalar(BaseComplexNDArray.java:1804)
at org.nd4j.linalg.api.complex.BaseComplexNDArray.copyFromReal(BaseComplexNDArray.java:545)
at org.nd4j.linalg.api.complex.BaseComplexNDArray.<init>(BaseComplexNDArray.java:159)
at org.nd4j.linalg.api.complex.BaseComplexNDArray.<init>(BaseComplexNDArray.java:167)
at org.nd4j.linalg.cpu.nativecpu.complex.ComplexNDArray.<init>(ComplexNDArray.java:104)
at org.nd4j.linalg.cpu.nativecpu.CpuNDArrayFactory.createComplex(CpuNDArrayFactory.java:166)
at org.nd4j.linalg.factory.Nd4j.createComplex(Nd4j.java:3345)
at org.nd4j.linalg.convolution.DefaultConvolutionInstance.convn(DefaultConvolutionInstance.java:116)
at org.nd4j.linalg.convolution.BaseConvolution.convn(BaseConvolution.java:66)
at com.example.demo.Main.testing(Main.java:41)
at com.example.demo.Main.main(Main.java:34)

Answer 1

这有点棘手，因为ND4j目前不支持数学卷积。你必须制定自己的实施方案。

double[] rawData = {12,10,15,12,10,11,15,12,12};
INDArray data = Nd4j.create(rawData);
double[] rawFilter = {1.0 / 2, 0, 1.0 / 2};
INDArray filter = Nd4j.create(rawFilter);        
Nd4jConv1d convolution = new Nd4jConv1d(1, 1, (int) filter.shape()[1], 1, 0);
INDArray output = convolution.forward(data, filter);

如图所示：https://github.com/deeplearning4j/deeplearning4j/blob/af7155d61dc810d3e7139f15f98810e0255b2e17/arbiter/arbiter-deeplearning4j/src/test/java/org/deeplearning4j/arbiter/multilayernetwork/MNISTOptimizationTest.java

注意：您需要一个额外的类Nd4jConv1d。去回购获得