public void TestGpuLayers() { var dataSet = new TestDataSet <float>(3, 4, 5, 10); Console.WriteLine("Testing softmax forward"); var softmaxLayer = new SoftMaxLayer <float>(dataSet.InputSize); TestLayerForward(softmaxLayer, dataSet, dataSet.InputSize); Console.WriteLine("Testing linear forward"); var linLayer = new LinearLayer <float>(dataSet.InputSize, dataSet.TargetSize, new RandomMatrixInitializer <float>()); TestLayerForward(linLayer, dataSet); Console.WriteLine("Testing GRU forward"); var gruLayer = new GruLayer <float>(dataSet.InputSize, dataSet.TargetSize, new ProportionalRandomMatrixInitializer <float>(), new ProportionalRandomMatrixInitializer <float>(), new RandomMatrixInitializer <float>()); TestLayerForward(gruLayer, dataSet); }
private unsafe void TestLayerForward(LayerBase <float> layer, TestDataSet <float> dataSet, int?outSize = null) { }