View source on GitHub
|
Computes the total size of instances in a Tensor over the whole dataset.
tft.size(
x, reduce_instance_dims=True, name=None
)
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.
|