Math Ops

Math Ops#

abs

Computes the element-wise absolute value of x.

add

Computes the element-wise sum of x and y.

cdiv

Computes the ceiling division of x by div

ceil

Computes the element-wise ceil of x.

clamp

Clamps the input tensor x within the range [min, max].

cos

Computes the element-wise cosine of x.

div

Element-wise division of x by y, the / operator on tensors.

div_rn

Computes the element-wise precise division (rounding to nearest wrt the IEEE standard) of x and y.

erf

Computes the element-wise error function of x.

exp

Computes the element-wise exponential of x.

exp2

Computes the element-wise exponential (base 2) of x.

fdiv

Computes the element-wise fast division of x and y.

floordiv

Element-wise integer division of x by y, the // operator on tensors.

floor

Computes the element-wise floor of x.

fma

Computes the element-wise fused multiply-add of x, y, and z.

log

Computes the element-wise natural logarithm of x.

log2

Computes the element-wise logarithm (base 2) of x.

maximum

Computes the element-wise maximum of x and y.

minimum

Computes the element-wise minimum of x and y.

mod

Element-wise remainder of x divided by y, the % operator on tensors.

mul

Computes the element-wise product of x and y.

neg

Element-wise negation, the unary - operator on tensors.

rsqrt

Computes the element-wise inverse square root of x.

sigmoid

Computes the element-wise sigmoid of x.

sin

Computes the element-wise sine of x.

softmax

Computes the element-wise softmax of x.

sqrt

Computes the element-wise fast square root of x.

sqrt_rn

Computes the element-wise precise square root (rounding to nearest wrt the IEEE standard) of x.

sub

Computes the element-wise difference of x and y.

umulhi

Computes the element-wise most significant N bits of the 2N-bit product of x and y.