TensorFlow.NET/tools/TensorFlowNET.Console/SimpleRnnTest.cs at master · ShuchenQ/TensorFlow.NET · GitHub

Repository navigation

26 lines (23 loc) · 821 Bytes

using System;

using System.Collections.Generic;

using System.Text;

using Tensorflow.Keras;

using Tensorflow.NumPy;

using static Tensorflow.Binding;

using static Tensorflow.KerasApi;

namespace Tensorflow

{

public class SimpleRnnTest

{

public void Run()

{

var inputs = np.random.random((6, 10, 8)).astype(np.float32);

//var simple_rnn = tf.keras.layers.SimpleRNN(4);

//var output = simple_rnn.Apply(inputs); // The output has shape `[32, 4]`.

var simple_rnn = tf.keras.layers.SimpleRNN(4, return_sequences: true, return_state: true);

// whole_sequence_output has shape `[32, 10, 4]`.

// final_state has shape `[32, 4]`.

var (whole_sequence_output, final_states) = simple_rnn.Apply(inputs);

}