tf.size
bookmark_borderbookmark
Stay organized with collections
Save and categorize content based on your preferences.
Returns the size of a tensor.
tf.size(
input, out_type=None, name=None
)
See also tf.shape
.
Returns a 0-D Tensor
representing the number of elements in input
of type out_type
. Defaults to tf.int32.
For example:
t = tf.constant([[[1, 1, 1], [2, 2, 2]], [[3, 3, 3], [4, 4, 4]]])
tf.size(t)
<tf.Tensor: shape=(), dtype=int32, numpy=12>
Args |
input
|
A Tensor or SparseTensor .
|
out_type
|
(Optional) The specified non-quantized numeric output type of the
operation. Defaults to tf.int32 . (Note: there is an experimental
flag, tf_shape_default_int64 that changes the default to tf.int64 .
This is an unsupported, experimental setting that causes known breakages.)
|
name
|
A name for the operation (optional).
|
Returns |
A Tensor of type out_type . Defaults to tf.int32 .
|
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-01-23 UTC.
[[["Easy to understand","easyToUnderstand","thumb-up"],["Solved my problem","solvedMyProblem","thumb-up"],["Other","otherUp","thumb-up"]],[["Missing the information I need","missingTheInformationINeed","thumb-down"],["Too complicated / too many steps","tooComplicatedTooManySteps","thumb-down"],["Out of date","outOfDate","thumb-down"],["Samples / code issue","samplesCodeIssue","thumb-down"],["Other","otherDown","thumb-down"]],["Last updated 2024-01-23 UTC."],[],[]]