Math Ops

Computes the absolute value of a tensor.

Returns the element-wise sum of a list of tensors.

Computes acos of x element-wise.

Computes inverse hyperbolic cosine of x element-wise.

Returns x + y element-wise.

Add all input tensors element wise.

Returns x + y element-wise.

Computes the "logical and" of elements across dimensions of a tensor.

Returns the argument of a complex number.

Computes the "logical or" of elements across dimensions of a tensor.

Returns the truth value of abs(x-y) < tolerance element-wise.

Returns the index with the largest value across dimensions of a tensor.

Returns the index with the smallest value across dimensions of a tensor.

Computes the trignometric inverse sine of x element-wise.

Computes inverse hyperbolic sine of x element-wise.

Computes the trignometric inverse tangent of x element-wise.

Computes arctangent of y/x element-wise, respecting signs of the arguments.

Computes inverse hyperbolic tangent of x element-wise.

Multiplies slices of two tensors in batches.

Multiplies slices of two tensors in batches.

Multiplies slices of two tensors in batches.

Compute the regularized incomplete beta integral \(I_x(a, b)\).

Counts the number of occurrences of each value in an integer array.

Bucketizes 'input' based on 'boundaries'.

Cast x of type SrcT to y of DstT.

Returns element-wise smallest integer not less than x.

Clips tensor values to a specified min and max.

Converts two real numbers to a complex number.

Computes the complex absolute value of a tensor.

Returns the complex conjugate of a complex number.

Computes cos of x element-wise.

Computes hyperbolic cosine of x element-wise.

Compute the pairwise cross product.

Compute the cumulative product of the tensor x along axis.

Compute the cumulative sum of the tensor x along axis.

Counts the number of occurrences of each value in an integer array.

Computes Psi, the derivative of Lgamma (the log of the absolute value of.

Returns x / y element-wise.

Returns 0 if the denominator is zero.

Returns the truth value of (x == y) element-wise.

Computes the Gauss error function of x element-wise.

Computes the complementary error function of x element-wise.

TODO: add doc.

Computes the euclidean norm of elements across dimensions of a tensor.

Computes exponential of x element-wise.

Computes exp(x) - 1 element-wise.

Returns element-wise largest integer not greater than x.

Returns x // y element-wise.

Returns element-wise remainder of division.

Returns the truth value of (x > y) element-wise.

Returns the truth value of (x >= y) element-wise.

Return histogram of values.

Compute the lower regularized incomplete Gamma function P(a, x).

Compute the upper regularized incomplete Gamma function Q(a, x).

Returns the imaginary part of a complex number.

Computes the reciprocal of x element-wise.

Returns which elements of x are finite.

Returns which elements of x are Inf.

Returns which elements of x are NaN.

Returns the truth value of (x < y) element-wise.

Returns the truth value of (x <= y) element-wise.

Computes the log of the absolute value of Gamma(x) element-wise.

Computes natural logarithm of x element-wise.

Computes natural logarithm of (1 + x) element-wise.

Returns the truth value of x AND y element-wise.

Returns the truth value of NOT x element-wise.

Returns the truth value of x OR y element-wise.

Multiply the matrix "a" by the matrix "b".

Computes the maximum of elements across dimensions of a tensor.

Returns the max of x and y (i.e.

Computes the mean of elements across dimensions of a tensor.

Computes the minimum of elements across dimensions of a tensor.

Returns the min of x and y (i.e.

Returns element-wise remainder of division.

Returns x * y element-wise.

Returns x * y element-wise.

TODO: add doc.

Computes numerical negative value element-wise.

Returns the next representable value of x1 in the direction of x2, element-wise.

Returns the truth value of (x != y) element-wise.

Compute the polygamma function \(\psi^{(n)}(x)\).

Computes the power of one value to another.

Computes the product of elements across dimensions of a tensor.

Convert the quantized 'input' tensor into a lower-precision 'output', using the.

Returns x + y element-wise, working on quantized buffers.

Perform a quantized matrix multiplication of a by the matrix b.

Returns x * y element-wise, working on quantized buffers.

Counts the number of occurrences of each value in an integer array.

Creates a sequence of numbers.

Returns the real part of a complex number.

Returns x / y element-wise for real types.

Computes the reciprocal of x element-wise.

Computes a range that covers the actual values present in a quantized tensor.

Converts the quantized input tensor into a lower-precision output.

Returns element-wise integer closest to x.

Rounds the values of a tensor to the nearest integer, element-wise.

Computes reciprocal of square root of x element-wise.

Computes the maximum along segments of a tensor.

Computes the maximum along segments of a tensor.

Computes the mean along segments of a tensor.

Computes the minimum along segments of a tensor.

Computes the minimum along segments of a tensor.

Computes the product along segments of a tensor.

Computes the product along segments of a tensor.

Computes the sum along segments of a tensor.

Computes the sum along segments of a tensor.

TODO: add doc.

Computes sigmoid of x element-wise.

Returns an element-wise indication of the sign of a number.

Computes sine of x element-wise.

Computes hyperbolic sine of x element-wise.

Counts the number of occurrences of each value in an integer array.

Multiply matrix "a" by matrix "b".

Computes the mean along sparse segments of a tensor.

Computes gradients for SparseSegmentMean.

Computes the mean along sparse segments of a tensor.

Computes the sum along sparse segments of a tensor divided by the sqrt of N.

Computes gradients for SparseSegmentSqrtN.

Computes the sum along sparse segments of a tensor divided by the sqrt of N.

Computes the sum along sparse segments of a tensor.

Computes gradients for SparseSegmentSum.

Computes the sum along sparse segments of a tensor.

Computes square root of x element-wise.

Computes square of x element-wise.

Returns conj(x - y)(x - y) element-wise.

Returns x - y element-wise.

Computes the sum of elements across dimensions of a tensor.

Computes tan of x element-wise.

Computes hyperbolic tangent of x element-wise.

Returns x / y element-wise, rounded towards zero.

Returns element-wise remainder of division.

Computes the maximum along segments of a tensor.

Computes the minimum along segments of a tensor.

Computes the product along segments of a tensor.

Computes the sum along segments of a tensor.

Selects elements from x or y, depending on condition.

Returns 0 if x == 0, and x / y otherwise, elementwise.

Returns 0 if x == 0, and x * log1p(y) otherwise, elementwise.

Returns 0 if x == 0, and x * log(y) otherwise, elementwise.

Compute the Hurwitz zeta function \(\zeta(x, q)\).