Browse Source

add tf.atan

tags/v0.9
Oceania2018 6 years ago
parent
commit
0677a1cbf0
2 changed files with 16 additions and 0 deletions
  1. +9
    -0
      src/TensorFlowNET.Core/APIs/tf.math.cs
  2. +7
    -0
      src/TensorFlowNET.Core/Operations/gen_math_ops.cs

+ 9
- 0
src/TensorFlowNET.Core/APIs/tf.math.cs View File

@@ -30,6 +30,15 @@ namespace Tensorflow
public static Tensor add(Tensor a, Tensor b)
=> gen_math_ops.add(a, b);

/// <summary>
/// Computes atan of x element-wise.
/// </summary>
/// <param name="x"></param>
/// <param name="name"></param>
/// <returns></returns>
public static Tensor atan(Tensor x, string name = null)
=> gen_math_ops.atan(x, name);

public static Tensor arg_max(Tensor input, int dimension, TF_DataType output_type = TF_DataType.TF_INT64, string name = null)
=> gen_math_ops.arg_max(input, dimension, output_type: output_type, name: name);



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

@@ -79,6 +79,13 @@ namespace Tensorflow
return _op.outputs[0];
}
public static Tensor atan(Tensor x, string name = null)
{
var _op = _op_def_lib._apply_op_helper("Atan", name, args: new { x });
return _op.outputs[0];
}
public static Tensor squared_difference(Tensor x, Tensor y, string name = null)
{
var _op = _op_def_lib._apply_op_helper("SquaredDifference", name, args: new { x, y, name });


Loading…
Cancel
Save