tf.where( condition, x=None, y=None, name=None )
Return the elements, either from
y, depending on the
y are None, then this operation returns the coordinates of true elements of
condition. The coordinates are returned in a 2-D tensor where the first dimension (rows) represents the number of true elements, and the second dimension (columns) represents the coordinates of the true elements. Keep in mind, the shape of the output tensor can vary depending on how many true values there are in input. Indices are output in row-major order.
If both non-None,
y must have the same shape. The
condition tensor must be a scalar if
y are scalar. If
y are vectors of higher rank, then
condition must be either a vector with size matching the first dimension of
x, or must have the same shape as
condition tensor acts as a mask that chooses, based on the value at each element, whether the corresponding element / row in the output should be taken from
x (if true) or
y (if false).
condition is a vector and
y are higher rank matrices, then it chooses which row (outer dimension) to copy from
condition has the same shape as
y, then it chooses which element to copy from
x: A Tensor which may have the same shape as
conditionis rank 1,
xmay have higher rank, but its first dimension must match the size of
tensorwith the same shape and type as
name: A name of the operation (optional)
Tensor with the same type and shape as
y if they are non-None. A
Tensor with shape
ValueError: When exactly one of
© 2018 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.