$OpDocMathNotEqual
$OpDocMathNotEqual
Result as a new tensor.
$OpDocMathMod
$OpDocMathMod
Result as a new tensor.
$OpDocMathLogicalAnd
$OpDocMathLogicalAnd
Result as a new tensor.
$OpDocMathMultiply
$OpDocMathMultiply
Result as a new tensor.
$OpDocMathPow
$OpDocMathPow
Result as a new tensor.
$OpDocMathAdd
$OpDocMathAdd
Result as a new tensor.
$OpDocMathSubtract
$OpDocMathSubtract
Result as a new tensor.
$OpDocMathDivide
$OpDocMathDivide
Result as a new tensor.
$OpDocMathLess
$OpDocMathLess
Result as a new tensor.
$OpDocMathLessEqual
$OpDocMathLessEqual
Result as a new tensor.
$OpDocMathEqual
$OpDocMathEqual
Result as a new tensor.
$OpDocMathGreater
$OpDocMathGreater
Result as a new tensor.
$OpDocMathGreaterEqual
$OpDocMathGreaterEqual
Result as a new tensor.
$OpDocMathPow
$OpDocMathPow
Result as a new tensor.
$OpDocMathAbs
$OpDocMathAbs
Result as a new tensor.
$OpDocMathAcos
$OpDocMathAcos
Result as a new tensor.
$OpDocMathAcosh
$OpDocMathAcosh
Result as a new tensor.
$OpDocMathAdd
$OpDocMathAdd
Result as a new tensor.
$OpDocMathAll
$OpDocMathAll
Integer tensor containing the axes to reduce. If null
, then all axes are reduced.
If true
, retain the reduced axes.
Result as a new tensor.
$OpDocMathAngle
$OpDocMathAngle
Result as a new tensor.
$OpDocMathAny
$OpDocMathAny
Integer tensor containing the axes to reduce. If null
, then all axes are reduced.
If true
, retain the reduced axes.
Result as a new tensor.
$OpDocMathApproximatelyEqual
$OpDocMathApproximatelyEqual
Result as a new tensor.
$OpDocMathArgmax
$OpDocMathArgmax
Integer tensor containing the axes to reduce. If null
, then all axes are reduced.
Data type for the output tensor. Must be INT32
or INT64
.
Result as a new tensor.
$OpDocMathArgmin
$OpDocMathArgmin
Integer tensor containing the axes to reduce. If null
, then all axes are reduced.
Data type for the output tensor. Must be INT32
or INT64
.
Result as a new tensor.
$OpDocMathAsin
$OpDocMathAsin
Result as a new tensor.
$OpDocMathAsinh
$OpDocMathAsinh
Result as a new tensor.
$OpDocMathAtan
$OpDocMathAtan
Result as a new tensor.
$OpDocMathAtan2
$OpDocMathAtan2
Result as a new tensor.
$OpDocMathAtanh
$OpDocMathAtanh
Result as a new tensor.
$OpDocMathBinCount
$OpDocMathBinCount
If not null
, this tensor must have the same shape as input
. For each value in input
, the
corresponding bin count will be incremented by the corresponding weight instead of 1
.
If not null
, this ensures the output has length at least minLength
, padding with zeros at
the end, if necessary.
If not null
, this skips values in input
that are equal or greater than maxLength
,
ensuring that the output has length at most maxLength
.
If weights
is null
, this determines the data type used for the output tensor (i.e., the
tensor containing the bin counts).
Result as a new tensor.
$OpDocMathBitcast
$OpDocMathBitcast
Target data type.
Result as a new tensor.
$OpDocMathBucketize
$OpDocMathBucketize
Sorted sequence of Float
s specifying the boundaries of the buckets.
Result as a new tensor.
$OpDocMathCast
$OpDocMathCast
Target data type.
Result as a new tensor.
$OpDocMathCeil
$OpDocMathCeil
Result as a new tensor.
$OpDocMathConjugate
$OpDocMathConjugate
Result as a new tensor.
$OpDocMathCos
$OpDocMathCos
Result as a new tensor.
$OpDocMathCosh
$OpDocMathCosh
Result as a new tensor.
$OpDocMathCountNonZero
$OpDocMathCountNonZero
Integer tensor containing the axes to reduce. If null
, then all axes are reduced.
If true
, retain the reduced axes.
Result as a new tensor.
$OpDocMathCross
$OpDocMathCross
Result as a new tensor.
$OpDocMathCumprod
$OpDocMathCumprod
INT32 tensor containing the axis along which to perform the cumulative product.
Boolean value indicating whether to perform an exclusive cumulative product.
Boolean value indicating whether to perform a reverse cumulative product.
Result as a new tensor.
$OpDocMathCumsum
$OpDocMathCumsum
INT32 tensor containing the axis along which to perform the cumulative sum.
Boolean value indicating whether to perform an exclusive cumulative sum.
Boolean value indicating whether to perform a reverse cumulative sum.
Result as a new tensor.
$OpDocMathDiag
$OpDocMathDiag
Result as a new tensor.
$OpDocMathDiagPart
$OpDocMathDiagPart
Result as a new tensor.
$OpDocMathDigamma
$OpDocMathDigamma
Result as a new tensor.
$OpDocMathDivide
$OpDocMathDivide
Result as a new tensor.
$OpDocMathEqual
$OpDocMathEqual
Result as a new tensor.
$OpDocMathErfc
$OpDocMathErfc
Result as a new tensor.
$OpDocMathErf
$OpDocMathErf
Result as a new tensor.
$OpDocMathExp
$OpDocMathExp
Result as a new tensor.
$OpDocMathExpm1
$OpDocMathExpm1
Result as a new tensor.
$OpDocMathFloor
$OpDocMathFloor
Result as a new tensor.
$OpDocMathFloorMod
$OpDocMathFloorMod
Result as a new tensor.
$OpDocMathGreater
$OpDocMathGreater
Result as a new tensor.
$OpDocMathGreaterEqual
$OpDocMathGreaterEqual
Result as a new tensor.
$OpDocMathIgamma
$OpDocMathIgamma
Result as a new tensor.
$OpDocMathIgammac
$OpDocMathIgammac
Result as a new tensor.
$OpDocMathImag
$OpDocMathImag
Result as a new tensor.
$OpDocMathIsFinite
$OpDocMathIsFinite
Result as a new tensor.
$OpDocMathIsInf
$OpDocMathIsInf
Result as a new tensor.
$OpDocMathIsNaN
$OpDocMathIsNaN
Result as a new tensor.
$OpDocMathLess
$OpDocMathLess
Result as a new tensor.
$OpDocMathLessEqual
$OpDocMathLessEqual
Result as a new tensor.
$OpDocMathLog
$OpDocMathLog
Result as a new tensor.
$OpDocMathLog1p
$OpDocMathLog1p
Result as a new tensor.
$OpDocMathLogGamma
$OpDocMathLogGamma
Result as a new tensor.
$OpDocMathLogSumExp
$OpDocMathLogSumExp
Integer tensor containing the axes to reduce. If null
, then all axes are reduced.
If true
, retain the reduced axes.
Result as a new tensor.
$OpDocMathLogicalAnd
$OpDocMathLogicalAnd
Result as a new tensor.
$OpDocMathLogicalNot
$OpDocMathLogicalNot
Result as a new tensor.
$OpDocMathLogicalOr
$OpDocMathLogicalOr
Result as a new tensor.
$OpDocMathLogicalXOr
$OpDocMathLogicalXOr
Result as a new tensor.
$OpDocMathMatmul
$OpDocMathMatmul
Output to multiply with, with data type one of: BFLOAT16
, FLOAT16
, FLOAT32
, FLOAT64
,
INT32
, COMPLEX64
, COMPLEX128
.
If true
, this tensor is transposed before the multiplication.
If true
, other
is transposed before the multiplication.
If true
, this tensor is conjugated before the multiplication.
If true
, other
is conjugated before the multiplication.
If true
, this tensor is treated as a sparse matrix (i.e., it is assumed it contains many
zeros).
If true
, other
is treated as a sparse matrix (i.e., it is assumed it contains many
zeros).
Result as a new tensor.
$OpDocMathMatrixBandPart
$OpDocMathMatrixBandPart
Scalar INT64
tensor that contains the number of sub-diagonals to keep. If negative,
the entire lower triangle is kept.
Scalar INT64
tensor that contains the number of super-diagonals to keep. If negative,
the entire upper triangle is kept.
Result as a new tensor containing the expected banded tensor and has rank K
and same shape as input
.
$OpDocMathMatrixDiag
$OpDocMathMatrixDiag
Result as a new tensor with rank equal to K + 1
and shape equal to the shape of diagonal
, with its
last dimension duplicated.
$OpDocMathMatrixDiagPart
$OpDocMathMatrixDiagPart
Result as a new tensor containing the diagonal(s) and having shape equal to
input.shape[:-2] + [min(input.shape[-2:])]
.
$OpDocMathMatrixSetDiag
$OpDocMathMatrixSetDiag
Rank-K
tensor, where K >= 1
.
Result as a new tensor with rank equal to K + 1
and shape equal to the shape of input
.
$OpDocMathMax
$OpDocMathMax
Integer tensor containing the axes to reduce. If null
, then all axes are reduced.
If true
, retain the reduced axes.
Result as a new tensor.
$OpDocMathMaximum
$OpDocMathMaximum
Result as a new tensor.
$OpDocMathMean
$OpDocMathMean
Integer tensor containing the axes to reduce. If null
, then all axes are reduced.
If true
, retain the reduced axes.
Result as a new tensor.
$OpDocMathMin
$OpDocMathMin
Integer tensor containing the axes to reduce. If null
, then all axes are reduced.
If true
, retain the reduced axes.
Result as a new tensor.
$OpDocMathMinimum
$OpDocMathMinimum
Result as a new tensor.
$OpDocMathMod
$OpDocMathMod
Result as a new tensor.
$OpDocMathMultiply
$OpDocMathMultiply
Result as a new tensor.
$OpDocMathNegate
$OpDocMathNegate
Result as a new tensor.
$OpDocMathNotEqual
$OpDocMathNotEqual
Result as a new tensor.
$OpDocMathPolygamma
$OpDocMathPolygamma
Result as a new tensor.
$OpDocMathPow
$OpDocMathPow
Result as a new tensor.
$OpDocMathProd
$OpDocMathProd
Integer tensor containing the axes to reduce. If null
, then all axes are reduced.
If true
, retain the reduced axes.
Result as a new tensor.
$OpDocMathReal
$OpDocMathReal
Result as a new tensor.
$OpDocMathRealDivide
$OpDocMathRealDivide
Result as a new tensor.
$OpDocMathReciprocal
$OpDocMathReciprocal
Result as a new tensor.
$OpDocMathRound
$OpDocMathRound
Result as a new tensor.
$OpDocMathRoundInt
$OpDocMathRoundInt
Result as a new tensor.
$OpDocMathRsqrt
$OpDocMathRsqrt
Result as a new tensor.
$OpDocMathSegmentMax
$OpDocMathSegmentMean
$OpDocMathSegmentMin
$OpDocMathSegmentProd
$OpDocMathSegmentSum
$OpDocMathSigmoid
$OpDocMathSigmoid
Result as a new tensor.
$OpDocMathSign
$OpDocMathSign
Result as a new tensor.
$OpDocMathSin
$OpDocMathSin
Result as a new tensor.
$OpDocMathSinh
$OpDocMathSinh
Result as a new tensor.
$OpDocMathSparseSegmentMean
$OpDocMathSparseSegmentMean
$OpDocMathSparseSegmentSum
$OpDocMathSparseSegmentSum
$OpDocMathSparseSegmentSumSqrtN
$OpDocMathSparseSegmentSumSqrtN
$OpDocMathSqrt
$OpDocMathSqrt
Result as a new tensor.
$OpDocMathSquare
$OpDocMathSquare
Result as a new tensor.
$OpDocMathSquaredDifference
$OpDocMathSquaredDifference
Result as a new tensor.
$OpDocMathSubtract
$OpDocMathSubtract
Result as a new tensor.
$OpDocMathSum
$OpDocMathSum
Integer tensor containing the axes to reduce. If null
, then all axes are reduced.
If true
, retain the reduced axes.
Result as a new tensor.
$OpDocMathTan
$OpDocMathTan
Result as a new tensor.
$OpDocMathTanh
$OpDocMathTanh
Result as a new tensor.
$OpDocMathTensorDot
$OpDocMathTensorDot
Tensor to contract with.
Axes to contract in this tensor.
Axes to contract in other
.
Name for the created ops.
Created op output.
$OpDocMathTensorDot
$OpDocMathTensorDot
Tensor to contract with.
Axes to contract in this tensor.
Axes to contract in other
.
Created op output.
$OpDocMathTensorDot
$OpDocMathTensorDot
Tensor to contract with.
Number of axes to contract.
Name for the created ops.
Created op output.
$OpDocMathTensorDot
$OpDocMathTensorDot
Tensor to contract with.
Number of axes to contract.
Created op output.
Dynamic version (i.e., where axes
and axesOther
may be symbolic tensors) of the tensorDot
op.
Dynamic version (i.e., where axes
and axesOther
may be symbolic tensors) of the tensorDot
op.
$OpDocMathTensorDot
Tensor to contract with.
Axes to contract in this tensor.
Axes to contract in other
.
Name for the created ops.
Created op output.
Dynamic version (i.e., where axes
and axesOther
may be symbolic tensors) of the tensorDot
op.
Dynamic version (i.e., where axes
and axesOther
may be symbolic tensors) of the tensorDot
op.
$OpDocMathTensorDot
Tensor to contract with.
Axes to contract in this tensor.
Axes to contract in other
.
Created op output.
Dynamic version (i.e., where numAxes
may be a symbolic tensor) of the tensorDot
op.
Dynamic version (i.e., where numAxes
may be a symbolic tensor) of the tensorDot
op.
$OpDocMathTensorDot
Tensor to contract with.
Number of axes to contract.
Name for the created ops.
Created op output.
Dynamic version (i.e., where numAxes
may be a symbolic tensor) of the tensorDot
op.
Dynamic version (i.e., where numAxes
may be a symbolic tensor) of the tensorDot
op.
$OpDocMathTensorDot
Tensor to contract with.
Number of axes to contract.
Created op output.
$OpDocMathTrace
$OpDocMathTrace
Result as a new tensor.
$OpDocMathTruncateDivide
$OpDocMathTruncateDivide
Result as a new tensor.
$OpDocMathTruncateMod
$OpDocMathTruncateMod
Result as a new tensor.
$OpDocMathLogicalNot
$OpDocMathLogicalNot
Result as a new tensor.
$OpDocMathNegate
$OpDocMathNegate
Result as a new tensor.
$OpDocMathUnsortedSegmentMax
$OpDocMathUnsortedSegmentMean
$OpDocMathUnsortedSegmentMin
$OpDocMathUnsortedSegmentProd
$OpDocMathUnsortedSegmentSqrtN
$OpDocMathUnsortedSegmentSum
$OpDocMathZerosFraction
$OpDocMathZerosFraction
Result as a new tensor, with FLOAT32
data type.
$OpDocMathZeta
$OpDocMathZeta
Result as a new tensor.
$OpDocMathLogicalOr
$OpDocMathLogicalOr
Result as a new tensor.