TF 2.0 is out! Get hands-on practice at TF World, Oct 28-31. Use code TF20 for 20% off select passes. Register now

tft.size

tft.size(
    x,
    reduce_instance_dims=True,
    name=None
)

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

Args:

  • x: A Tensor or SparseTensor.
  • 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.