diff options
author | 2017-10-02 15:27:14 -0700 | |
---|---|---|
committer | 2017-10-02 15:30:37 -0700 | |
commit | 320a824ba358856a9d88779b49f6810d434c8d27 (patch) | |
tree | 65232f920b63032e2616d5616f7358249ccad229 /tensorflow/go | |
parent | c5123fe53da9fb2f3ef5b390e7992a679d994d36 (diff) |
Go: Update generated wrapper functions for TensorFlow ops.
PiperOrigin-RevId: 170763068
Diffstat (limited to 'tensorflow/go')
-rw-r--r-- | tensorflow/go/op/wrappers.go | 30 |
1 files changed, 29 insertions, 1 deletions
diff --git a/tensorflow/go/op/wrappers.go b/tensorflow/go/op/wrappers.go index 8131d74342..09a509f21b 100644 --- a/tensorflow/go/op/wrappers.go +++ b/tensorflow/go/op/wrappers.go @@ -1412,7 +1412,7 @@ func BroadcastArgs(scope *Scope, s0 tf.Output, s1 tf.Output) (r0 tf.Output) { return op.Output(0) } -// Returns locations of true values in a boolean tensor. +// Returns locations of nonzero / true values in a tensor. // // This operation returns the coordinates of true elements in `input`. The // coordinates are returned in a 2-D tensor where the first dimension (rows) @@ -1444,6 +1444,34 @@ func BroadcastArgs(scope *Scope, s0 tf.Output, s1 tf.Output) (r0 tf.Output) { // [1, 0, 1], // [1, 1, 1], // [2, 1, 1]] +// +// # `input` tensor is [[[1.5, 0.0] +// # [-0.5, 0.0]] +// # [[0.0, 0.25] +// # [0.0, 0.75]] +// # [[0.0, 0.0] +// # [0.0, 0.01]]] +// # 'input' has 5 nonzero values, so output has 5 coordinates. +// # 'input' has rank of 3, so coordinates have three indices. +// where(input) ==> [[0, 0, 0], +// [0, 1, 0], +// [1, 0, 1], +// [1, 1, 1], +// [2, 1, 1]] +// +// # `input` tensor is [[[1.5 + 0.0j, 0.0 + 0.0j] +// # [0.0 + 0.5j, 0.0 + 0.0j]] +// # [[0.0 + 0.0j, 0.25 + 1.5j] +// # [0.0 + 0.0j, 0.75 + 0.0j]] +// # [[0.0 + 0.0j, 0.0 + 0.0j] +// # [0.0 + 0.0j, 0.01 + 0.0j]]] +// # 'input' has 5 nonzero magnitude values, so output has 5 coordinates. +// # 'input' has rank of 3, so coordinates have three indices. +// where(input) ==> [[0, 0, 0], +// [0, 1, 0], +// [1, 0, 1], +// [1, 1, 1], +// [2, 1, 1]] // ``` func Where(scope *Scope, input tf.Output) (index tf.Output) { if scope.Err() != nil { |