You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.

TensorTest.cs 1.2 kB

6 years ago
6 years ago
1234567891011121314151617181920212223242526272829303132333435363738394041
  1. using Microsoft.VisualStudio.TestTools.UnitTesting;
  2. using NumSharp.Core;
  3. using System;
  4. using System.Collections.Generic;
  5. using System.Linq;
  6. using System.Runtime.InteropServices;
  7. using System.Text;
  8. using Tensorflow;
  9. namespace TensorFlowNET.UnitTest
  10. {
  11. [TestClass]
  12. public class TensorTest
  13. {
  14. [TestMethod]
  15. public void TF_NewTensor()
  16. {
  17. var nd = np.array(1f, 2f, 3f, 4f, 5f, 6f).reshape(2, 3);
  18. var data = Marshal.AllocHGlobal(sizeof(float) * nd.size);
  19. Marshal.Copy(nd.Data<float>(), 0, data, nd.size);
  20. var handle = c_api.TF_NewTensor(TF_DataType.TF_FLOAT,
  21. nd.shape.Select(x => (long)x).ToArray(), // shape
  22. nd.ndim,
  23. data,
  24. (UIntPtr)(nd.size * sizeof(float)),
  25. tf.FreeTensorData,
  26. IntPtr.Zero);
  27. Assert.AreNotEqual(handle, IntPtr.Zero);
  28. var tensor = new Tensor(handle);
  29. Assert.AreEqual(tensor.dtype, TF_DataType.TF_FLOAT);
  30. Assert.AreEqual(tensor.ndim, nd.ndim);
  31. Assert.AreEqual(nd.shape[0], c_api.TF_Dim(handle, 0));
  32. Assert.AreEqual(nd.shape[1], c_api.TF_Dim(handle, 1));
  33. }
  34. }
  35. }

tensorflow框架的.NET版本,提供了丰富的特性和API,可以借此很方便地在.NET平台下搭建深度学习训练与推理流程。

Contributors (1)