@@ -50,7 +50,7 @@ public static Tensor add_n(Tensor[] inputs, string name = null)
5050 /// <param name="output_type"></param>
5151 /// <param name="name"></param>
5252 /// <returns></returns>
53- public static Tensor arg_max ( Tensor input , int dimension , TF_DataType output_type = TF_DataType . TF_INT64 , string name = null )
53+ public static Tensor arg_max ( Tensor input , Axis dimension , TF_DataType output_type = TF_DataType . TF_INT64 , string name = null )
5454 => tf . Context . ExecuteOp ( "ArgMax" , name , new ExecuteOpArgs ( input , dimension )
5555 . SetAttributes ( new { output_type } ) ) ;
5656
@@ -308,10 +308,7 @@ public static Tensor less_equal<Tx, Ty>(Tx x, Ty y, string name = null)
308308 public static Tensor log1p ( Tensor x , string name = null )
309309 => tf . Context . ExecuteOp ( "Log1p" , name , new ExecuteOpArgs ( x ) ) ;
310310
311- public static Tensor logical_and ( Tensor x , Tensor y , string name = null )
312- => tf . Context . ExecuteOp ( "LogicalAnd" , name , new ExecuteOpArgs ( x , y ) ) ;
313-
314- public static Tensor logical_and ( bool x , bool y , string name = null )
311+ public static Tensor logical_and < T > ( T x , T y , string name = null )
315312 => tf . Context . ExecuteOp ( "LogicalAnd" , name , new ExecuteOpArgs ( x , y ) ) ;
316313
317314 public static Tensor logical_not ( Tensor x , string name = null )
0 commit comments