tft.mean | TFX | TensorFlow (original) (raw)

tft.mean

Computes the mean of the values of a Tensor over the whole dataset.

tft.mean(
    x: common_types.TensorType,
    reduce_instance_dims: bool = True,
    name: Optional[str] = None,
    output_dtype: Optional[tf.DType] = None
) -> tf.Tensor

Used in the notebooks

Used in the tutorials
Preprocess data with TensorFlow Transform

Args
x	A Tensor, SparseTensor, or RaggedTensor. Its type must be floating point (float{16\|32
reduce_instance_dims	By default collapses the batch and instance dimensions to arrive at a single scalar output. If False, only collapses the batch dimension and outputs a vector of the same shape as the input.
name	(Optional) A name for this operation.
output_dtype	(Optional) If not None, casts the output tensor to this type.

Returns
A Tensor containing the mean. If x is floating point, the mean will have the same type as x. If x is integral, the output is cast to float32. NaNs and infinite input values are ignored.

Raises
TypeError	If the type of x is not supported.

Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 4.0 License, and code samples are licensed under the Apache 2.0 License. For details, see the Google Developers Site Policies. Java is a registered trademark of Oracle and/or its affiliates.

Last updated 2024-11-01 UTC.