diff --git a/src/TensorFlowNET.Core/APIs/tf.array.cs b/src/TensorFlowNET.Core/APIs/tf.array.cs index 9dc72165..4727a00f 100644 --- a/src/TensorFlowNET.Core/APIs/tf.array.cs +++ b/src/TensorFlowNET.Core/APIs/tf.array.cs @@ -66,6 +66,17 @@ namespace Tensorflow public Tensor fill(Tensor dims, T value, string name = null) => gen_array_ops.fill(dims, value, name: name); + /// + /// Gather slices from params axis axis according to indices. + /// + /// + /// + /// + /// + /// + public Tensor gather(Tensor @params, Tensor indices, string name = null, int axis = 0) + => array_ops.gather(@params, indices, name: name, axis: axis); + /// /// Return the elements, either from `x` or `y`, depending on the `condition`. ///