diff --git a/src/TensorFlowNET.Core/APIs/tf.math.cs b/src/TensorFlowNET.Core/APIs/tf.math.cs index f7f3ae61..ca047fdd 100644 --- a/src/TensorFlowNET.Core/APIs/tf.math.cs +++ b/src/TensorFlowNET.Core/APIs/tf.math.cs @@ -30,6 +30,15 @@ namespace Tensorflow public static Tensor add(Tensor a, Tensor b) => gen_math_ops.add(a, b); + /// + /// Computes atan of x element-wise. + /// + /// + /// + /// + 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); diff --git a/src/TensorFlowNET.Core/Operations/gen_math_ops.cs b/src/TensorFlowNET.Core/Operations/gen_math_ops.cs index b02aaee9..03cbee13 100644 --- a/src/TensorFlowNET.Core/Operations/gen_math_ops.cs +++ b/src/TensorFlowNET.Core/Operations/gen_math_ops.cs @@ -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 });