Browse Source

add tf.nn.lrn Local Response Normalization.

tags/v0.12
Oceania2018 6 years ago
parent
commit
8934c976c5
2 changed files with 40 additions and 0 deletions
  1. +15
    -0
      src/TensorFlowNET.Core/APIs/tf.nn.cs
  2. +25
    -0
      src/TensorFlowNET.Core/Operations/NnOps/gen_nn_ops.cs

+ 15
- 0
src/TensorFlowNET.Core/APIs/tf.nn.cs View File

@@ -148,6 +148,21 @@ namespace Tensorflow
}); });
} }


/// <summary>
/// Local Response Normalization.
/// </summary>
/// <param name="input"></param>
/// <param name="depth_radius"></param>
/// <param name="bias"></param>
/// <param name="alpha"></param>
/// <param name="beta"></param>
/// <param name="name"></param>
/// <returns></returns>
public Tensor lrn(Tensor input, int depth_radius = 5, int bias = 1,
int alpha = 1, float beta = 0.5f, string name = null)
=> gen_nn_ops.local_response_normalization(input, depth_radius: depth_radius, bias: bias,
alpha: alpha, beta: beta, name: name);

public rnn_cell_impl rnn_cell => new rnn_cell_impl(); public rnn_cell_impl rnn_cell => new rnn_cell_impl();


public Tensor softmax(Tensor logits, int axis = -1, string name = null) public Tensor softmax(Tensor logits, int axis = -1, string name = null)


+ 25
- 0
src/TensorFlowNET.Core/Operations/NnOps/gen_nn_ops.cs View File

@@ -181,6 +181,31 @@ namespace Tensorflow.Operations
return _op.outputs; return _op.outputs;
} }


/// <summary>
/// Local Response Normalization.
/// </summary>
/// <param name="input"></param>
/// <param name="depth_radius"></param>
/// <param name="bias"></param>
/// <param name="alpha"></param>
/// <param name="beta"></param>
/// <param name="name"></param>
/// <returns></returns>
public static Tensor local_response_normalization(Tensor input, int depth_radius = 5, int bias = 1,
int alpha = 1, float beta = 0.5f, string name = null)
{
var _op = _op_def_lib._apply_op_helper("LRN", name: name, args: new
{
input,
depth_radius,
bias,
alpha,
beta
});

return _op.output;
}

public static Tensor log_softmax(Tensor logits, string name = null) public static Tensor log_softmax(Tensor logits, string name = null)
{ {
var _op = _op_def_lib._apply_op_helper("LogSoftmax", name: name, args: new var _op = _op_def_lib._apply_op_helper("LogSoftmax", name: name, args: new


Loading…
Cancel
Save