tf.raw_ops.ExperimentalGroupByReducerDataset | TensorFlow v2.16.1 (original) (raw)

tf.raw_ops.ExperimentalGroupByReducerDataset

Stay organized with collections Save and categorize content based on your preferences.

Creates a dataset that computes a group-by on input_dataset.

View aliases

Compat aliases for migration

tf.compat.v1.raw_ops.ExperimentalGroupByReducerDataset

tf.raw_ops.ExperimentalGroupByReducerDataset(
    input_dataset,
    key_func_other_arguments,
    init_func_other_arguments,
    reduce_func_other_arguments,
    finalize_func_other_arguments,
    key_func,
    init_func,
    reduce_func,
    finalize_func,
    output_types,
    output_shapes,
    name=None
)

Creates a dataset that computes a group-by on input_dataset.

Args
input_dataset	A Tensor of type variant. A variant tensor representing the input dataset.
key_func_other_arguments	A list of Tensor objects. A list of tensors, typically values that were captured when building a closure for key_func.
init_func_other_arguments	A list of Tensor objects. A list of tensors, typically values that were captured when building a closure for init_func.
reduce_func_other_arguments	A list of Tensor objects. A list of tensors, typically values that were captured when building a closure for reduce_func.
finalize_func_other_arguments	A list of Tensor objects. A list of tensors, typically values that were captured when building a closure for finalize_func.
key_func	A function decorated with @Defun. A function mapping an element of input_dataset, concatenated with key_func_other_arguments to a scalar value of type DT_INT64.
init_func	A function decorated with @Defun. A function mapping a key of type DT_INT64, concatenated withinit_func_other_arguments to the initial reducer state.
reduce_func	A function decorated with @Defun. A function mapping the current reducer state and an element of input_dataset, concatenated with reduce_func_other_arguments to a new reducer state.
finalize_func	A function decorated with @Defun. A function mapping the final reducer state to an output element.
output_types	A list of tf.DTypes that has length >= 1.
output_shapes	A list of shapes (each a tf.TensorShape or list of ints) that has length >= 1.
name	A name for the operation (optional).

Returns
A Tensor of type variant.

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. Some content is licensed under the numpy license.

Last updated 2024-04-26 UTC.