Browse Source

add overload for TF_AllocateTensor.

tags/v0.12
Oceania2018 6 years ago
parent
commit
ce7d5fed5b
1 changed files with 3 additions and 0 deletions
  1. +3
    -0
      src/TensorFlowNET.Core/Tensors/c_api.tensor.cs

+ 3
- 0
src/TensorFlowNET.Core/Tensors/c_api.tensor.cs View File

@@ -35,6 +35,9 @@ namespace Tensorflow
[DllImport(TensorFlowLibName)] [DllImport(TensorFlowLibName)]
public static extern IntPtr TF_AllocateTensor(TF_DataType dtype, long[] dims, int num_dims, ulong len); public static extern IntPtr TF_AllocateTensor(TF_DataType dtype, long[] dims, int num_dims, ulong len);


[DllImport(TensorFlowLibName)]
public static extern IntPtr TF_AllocateTensor(TF_DataType dtype, long[] dims, int num_dims, UIntPtr len);

/// <summary> /// <summary>
/// returns the sizeof() for the underlying type corresponding to the given TF_DataType enum value. /// returns the sizeof() for the underlying type corresponding to the given TF_DataType enum value.
/// </summary> /// </summary>


Loading…
Cancel
Save