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.
|