Examples with UniformInitScheme org.nd4j.weightinit.impl.UniformInitScheme used on opensource projects

Example 1 with UniformInitScheme

use of org.nd4j.weightinit.impl.UniformInitScheme in project nd4j by deeplearning4j.

the class SameDiffTests method testLinearModule.

@Test
public void testLinearModule() {
    int nIn = 5;
    Linear linear = Linear.execBuilder().nIn(nIn).nOut(4).weightInitScheme(new UniformInitScheme('f', nIn)).biasWeightInitScheme(new ZeroInitScheme('f')).build();
    linear.exec(Nd4j.linspace(1, 20, 20).reshape(4, 5));
    assertEquals(1, linear.numOutputArguments());
}