Oceania2018 5 years ago
parent
commit
c10f610a89
2 changed files with 41 additions and 0 deletions
  1. +10
    -0
      src/TensorFlowNET.Core/APIs/tf.debugging.cs
  2. +31
    -0
      src/TensorFlowNET.Core/Operations/check_ops.cs

+ 10
- 0
src/TensorFlowNET.Core/APIs/tf.debugging.cs View File

@@ -40,5 +40,15 @@ namespace Tensorflow
message: message,
name: name);

public Tensor assert_greater_equal<T1, T2>(Tensor x,
Tensor y,
object[] data = null,
string message = null,
string name = null)
=> check_ops.assert_greater_equal(x,
y,
data: data,
message: message,
name: name);
}
}

+ 31
- 0
src/TensorFlowNET.Core/Operations/check_ops.cs View File

@@ -56,6 +56,37 @@ namespace Tensorflow
return control_flow_ops.Assert(condition, data);
});
}
public static Operation assert_greater_equal(Tensor x, Tensor y, object[] data = null, string message = null,
string name = null)
{
if (message == null)
message = "";

return tf_with(ops.name_scope(name, "assert_greater_equal", new {x, y, data}), delegate
{
x = ops.convert_to_tensor(x, name: "x");
y = ops.convert_to_tensor(y, name: "y");
string x_name = x.name;
string y_name = y.name;
if (data == null)
{
data = new object[]
{
message,
"Condition x >= y did not hold element-wise:",
$"x (%s) = {x_name}",
x,
$"y (%s) = {y_name}",
y
};
}

var condition = math_ops.reduce_all(gen_math_ops.greater_equal(x, y));
return control_flow_ops.Assert(condition, data);
});
}


public static Operation assert_positive(Tensor x, object[] data = null, string message = null, string name = null)
{


Loading…
Cancel
Save