View source on GitHub
|
Operations that use static values when possible.
Modules
dtype_util module: Utility functions for dtypes.
tensorshape_util module: Utility functions for TensorShape.
Functions
abs(...): Computes the absolute value of a tensor.
add(...): Returns x + y element-wise.
argmax(...): Returns the index with the largest value across axes of a tensor.
argmin(...): Returns the index with the smallest value across axes of a tensor.
argsort(...): Returns the indices of a tensor that give its sorted order along an axis.
broadcast_shape(...): Computes the shape of a broadcast.
broadcast_to(...): Broadcast an array for a compatible shape.
case(...): Like tf.case, except attempts to statically evaluate predicates.
cast(...): Casts a tensor to a new type.
ceil(...): Return the ceiling of the input, element-wise.
concat(...): Concatenates tensors along one dimension.
cond(...): Return either true_fn() if predicate pred is true else false_fn().
constant(...): Creates a constant tensor from a tensor-like object.
convert_to_shape_tensor(...): Converts the given value to a Tensor.
cumprod(...): Compute the cumulative product of the tensor x along axis.
cumsum(...): Compute the cumulative sum of the tensor x along axis.
dimension_size(...): Equivalent to shape(x)[idx], but robust to partially-known shapes.
equal(...): Returns the truth value of (x == y) element-wise.
expand_dims(...): Returns a tensor with a length 1 axis inserted at index axis.
expm1(...): Computes exp(x) - 1 element-wise.
eye(...): Construct an identity matrix, or a batch of matrices.
fill(...): Creates a tensor filled with a scalar value.
floor(...): Returns element-wise largest integer not greater than x.
gather(...): Gather slices from params axis axis according to indices. (deprecated arguments)
greater(...): Returns the truth value of (x > y) element-wise.
identity(...): Return a Tensor with the same shape and contents as input.
invert_permutation(...): Computes the inverse permutation of a tensor.
is_finite(...): Returns which elements of x are finite.
is_inf(...): Returns which elements of x are Inf.
is_nan(...): Returns which elements of x are NaN.
is_numpy(...): Returns true if x is a numpy object.
less(...): Returns the truth value of (x < y) element-wise.
linspace(...): Generates evenly-spaced values in an interval along a given axis.
log(...): Computes natural logarithm of x element-wise.
log1p(...): Computes natural logarithm of (1 + x) element-wise.
logical_and(...): Returns the truth value of x AND y element-wise.
logical_not(...): Returns the truth value of NOT x element-wise.
logical_or(...): Returns the truth value of x OR y element-wise.
maximum(...): Returns the max of x and y (i.e. x > y ? x : y) element-wise.
minimum(...): Returns the min of x and y (i.e. x < y ? x : y) element-wise.
nextafter(...): Returns the next representable value of x1 in the direction of x2, element-wise.
non_negative_axis(...): Make (possibly negatively indexed) axis argument non-negative.
not_equal(...): Returns the truth value of (x != y) element-wise.
one_hot(...): Returns a one-hot tensor.
ones(...): Creates a tensor with all elements set to one (1).
ones_like(...): Creates a tensor of all ones that has the same shape as the input.
pad(...): Pads a tensor.
pow(...): Computes the power of one value to another.
range(...): Creates a sequence of numbers.
rank(...): Returns the rank of a tensor.
rank_from_shape(...): Computes rank given a Tensor's shape.
reduce_all(...): Computes tf.math.logical_and of elements across dimensions of a tensor.
reduce_any(...): Computes tf.math.logical_or of elements across dimensions of a tensor.
reduce_max(...): Computes tf.math.maximum of elements across dimensions of a tensor.
reduce_min(...): Computes the tf.math.minimum of elements across dimensions of a tensor.
reduce_prod(...): Computes tf.math.multiply of elements across dimensions of a tensor.
reduce_sum(...): Computes the sum of elements across dimensions of a tensor.
repeat(...): Repeat elements of input.
reshape(...): Reshapes a tensor.
reverse(...): Reverses specific dimensions of a tensor.
round(...): Rounds the values of a tensor to the nearest integer, element-wise.
rsqrt(...): Computes reciprocal of square root of x element-wise.
setdiff1d(...): Compute set difference of elements in last dimension of a and b.
shape(...): Returns a tensor containing the shape of the input tensor.
shape_slice(...): Equivalent to shape(x)[slice_], but robust to partially-known shapes.
size(...): Returns the size of a tensor.
size0(...): Returns the size of the first dimension (0 if scalar).
slice(...): Extracts a slice from a tensor.
smart_where(...): As tf.where, but only calls x_fn/y_fn when condition not statically known.
sort(...): Sorts a tensor.
split(...): Splits a tensor value into a list of sub tensors.
sqrt(...): Computes element-wise square root of the input tensor.
stack(...): Stacks a list of rank-R tensors into one rank-(R+1) tensor.
tensor_scatter_nd_add(...): Adds sparse updates to an existing tensor according to indices.
tensor_scatter_nd_sub(...): Subtracts sparse updates from an existing tensor according to indices.
tensor_scatter_nd_update(...): Scatter updates into an existing tensor according to indices.
tile(...): Constructs a tensor by tiling a given tensor.
top_k(...): Finds values and indices of the k largest entries for the last dimension.
unique(...): Finds unique elements in a 1-D tensor.
unstack(...): Unpacks the given dimension of a rank-R tensor into rank-(R-1) tensors.
where(...): Returns the indices of non-zero elements, or multiplexes x and y.
zeros(...): Creates a tensor with all elements set to zero.
zeros_like(...): Creates a tensor with all elements set to zero.
Other Members | |
|---|---|
| JAX_MODE |
False
|