Unable to use more than 255 output neurons on the CPU backend

[bittere]

Hi.

Here is a simple neural network with an output of 2 neurons:

import { Cost, CPU, DenseLayer, Sequential, setupBackend, SigmoidLayer, tensor1D, tensor2D, } from "https://deno.land/x/netsaur/mod.ts";

await setupBackend(CPU);

const net = new Sequential({
    size: [4, 2],
    layers: [
      DenseLayer({ size: [3] }),
      SigmoidLayer(),
      DenseLayer({ size: [2] }),
      SigmoidLayer(),
    ],
    cost: Cost.MSE,
  });

const outputs = await net.predict(tensor1D([1, 0]));
console.log(outputs);

$ deno run -A --unstable test.ts

Tensor {                                                                                                                  
    shape: [ 2 ],
    data: Float32Array(2) [ 0.3955751657485962, 0.6075010895729065 ]
}

However, if I update the last layer to 256 (or more) neurons, this happens:

$ deno run -A --unstable test.ts

thread '<unnamed>' panicked at 'source slice length (256) does not match destination slice length (0)', rust\src\ffi.rs:74:17
note: run with `RUST_BACKTRACE=1` environment variable to display a backtrace
fatal runtime error: Rust cannot catch foreign exceptions

With 255 neurons or less, this error does not happen.

It also looks like this only happens when using the CPU backend, not the WASM backend. There is no error if I use the WASM backend.

I am running deno 1.37.1 on Windows 10.

Thanks in advance!

[load1n9]

Will look into it ASAP

[retraigo]

Can we close this now that #51 is merged?

[bittere]

Yes, it appears to be working now. Thanks!