Browse Source

add gen_array_ops.pack eager

tags/v0.20
Esther2013 Haiping 5 years ago
parent
commit
653427144a
3 changed files with 13 additions and 3 deletions
  1. +7
    -0
      src/TensorFlowNET.Core/Operations/array_ops.cs
  2. +6
    -0
      src/TensorFlowNET.Core/Operations/gen_array_ops.cs
  3. +0
    -3
      src/TensorFlowNET.Core/ops.cs

+ 7
- 0
src/TensorFlowNET.Core/Operations/array_ops.cs View File

@@ -178,6 +178,13 @@ namespace Tensorflow
return dtype;
}

/// <summary>
/// Converts the given list or tuple to a tensor by packing.
/// </summary>
/// <param name="list_or_tuple">A (possibly nested) list or tuple containing a tensor.</param>
/// <param name="dtype"></param>
/// <param name="name"></param>
/// <returns>A `tf.Tensor` with value equivalent to `list_or_tuple`.</returns>
public static Tensor _autopacking_helper(object[] list_or_tuple, TF_DataType dtype, string name)
{
var must_pack = false;


+ 6
- 0
src/TensorFlowNET.Core/Operations/gen_array_ops.cs View File

@@ -122,6 +122,12 @@ namespace Tensorflow

public static Tensor pack(Tensor[] values, int axis = 0, string name = null)
{
if(tf.context.executing_eagerly())
{
var _result = pywrap_tfe_src.TFE_Py_FastPathExecute(tf.context, tf.context.device_name, "Pack", name, null, values, "axis", axis);
return _result;
}

var _op = _op_def_lib._apply_op_helper("Pack", name: name, args: new { values, axis });

return _op.outputs[0];


+ 0
- 3
src/TensorFlowNET.Core/ops.cs View File

@@ -97,9 +97,6 @@ namespace Tensorflow
/// <returns></returns>
public static Tensor convert_to_tensor(object value, TF_DataType dtype = TF_DataType.DtInvalid, string name = null, TF_DataType preferred_dtype = TF_DataType.DtInvalid)
{
if (value is Tensor tensor)
return tensor;

return convert_to_tensor_v2(value, dtype, preferred_dtype, name);
}



Loading…
Cancel
Save