Browse Source

tf.gather #396

tags/v0.12
Oceania2018 6 years ago
parent
commit
5640bf0f45
1 changed files with 11 additions and 0 deletions
  1. +11
    -0
      src/TensorFlowNET.Core/APIs/tf.array.cs

+ 11
- 0
src/TensorFlowNET.Core/APIs/tf.array.cs View File

@@ -66,6 +66,17 @@ namespace Tensorflow
public Tensor fill<T>(Tensor dims, T value, string name = null) public Tensor fill<T>(Tensor dims, T value, string name = null)
=> gen_array_ops.fill(dims, value, name: name); => gen_array_ops.fill(dims, value, name: name);


/// <summary>
/// Gather slices from params axis axis according to indices.
/// </summary>
/// <param name="params"></param>
/// <param name="indices"></param>
/// <param name="name"></param>
/// <param name="axis"></param>
/// <returns></returns>
public Tensor gather(Tensor @params, Tensor indices, string name = null, int axis = 0)
=> array_ops.gather(@params, indices, name: name, axis: axis);

/// <summary> /// <summary>
/// Return the elements, either from `x` or `y`, depending on the `condition`. /// Return the elements, either from `x` or `y`, depending on the `condition`.
/// </summary> /// </summary>


Loading…
Cancel
Save