Use of Nx.Random.uniform_split/2
I think should be Nx.Random.uniform_split/4
, e.g.:
Nx.Random.uniform_split(new_key, 0, 1, shape: {})
|> NeuralNetwork.predict(w1, b1, w2, b2)
Use of Nx.Random.uniform_split/2
I think should be Nx.Random.uniform_split/4
, e.g.:
Nx.Random.uniform_split(new_key, 0, 1, shape: {})
|> NeuralNetwork.predict(w1, b1, w2, b2)