GitHub

tft.size

Computes the total size of instances in a Tensor over the whole dataset.

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

Args
`x`	A `Tensor`, `SparseTensor`, or `RaggedTensor`.
`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.

Returns
A `Tensor` of type int64.

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 2023-04-14 UTC.

Args