|
- using System;
- using Microsoft.VisualStudio.TestTools.UnitTesting;
- using NumSharp;
- using Tensorflow;
- using static Tensorflow.Binding;
-
- namespace TensorFlowNET.UnitTest.functional_ops_test
- {
- /// <summary>
- /// https://www.tensorflow.org/api_docs/python/tf/scan
- /// </summary>
- [TestClass]
- public class ScanTestCase
- {
- [Ignore("TODO")]
- [TestMethod]
- public void ScanForward()
- {
- var fn = new Func<Tensor, Tensor, Tensor>((a, x) => tf.add(a, x));
-
- var sess = tf.Session().as_default();
-
- var input = tf.placeholder(TF_DataType.TF_INT32, new TensorShape(6));
- var scan = functional_ops.scan(fn, input);
- sess.run(scan, (input, np.array(1,2,3,4,5,6))).Should().Be(np.array(1,3,6,10,15,21));
- }
-
- [Ignore("TODO")]
- [TestMethod]
- public void ScanReverse()
- {
- var fn = new Func<Tensor, Tensor, Tensor>((a, x) => tf.add(a, x));
-
- var sess = tf.Session().as_default();
-
- var input = tf.placeholder(TF_DataType.TF_INT32, new TensorShape(6));
- var scan = functional_ops.scan(fn, input, reverse:true);
- sess.run(scan, (input, np.array(1,2,3,4,5,6))).Should().Be(np.array(21,20,18,15,11,6));
- }
- }
- }
|