|
|
@@ -77,6 +77,9 @@ namespace Tensorflow |
|
|
|
public static Tensor argmax(Tensor input, Axis 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); |
|
|
|
|
|
|
|
public static Tensor argmin(Tensor input, Axis dimension, TF_DataType output_type = TF_DataType.TF_INT64, string name = null) |
|
|
|
=> gen_math_ops.arg_min(input, dimension, output_type: output_type, name: name); |
|
|
|
|
|
|
|
public static Tensor round(Tensor x, string name = null) |
|
|
|
{ |
|
|
|
x = ops.convert_to_tensor(x, name: "x"); |
|
|
|