tf.compat.v1.size
Stay organized with collections
Save and categorize content based on your preferences.
Returns the size of a tensor.
tf.compat.v1.size(
input, name=None, out_type=None
)
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) # 12
Args |
input
|
A Tensor or SparseTensor .
|
name
|
A name for the operation (optional).
|
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.)
|
Returns |
A Tensor of type out_type . Defaults to tf.int32 .
|
Equivalent to np.size()
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 2023-10-06 UTC.
[null,null,["Last updated 2023-10-06 UTC."],[],[],null,["# tf.compat.v1.size\n\n\u003cbr /\u003e\n\n|-----------------------------------------------------------------------------------------------------------------------------|\n| [View source on GitHub](https://github.com/tensorflow/tensorflow/blob/v2.14.0/tensorflow/python/ops/array_ops.py#L811-L847) |\n\nReturns the size of a tensor. \n\n tf.compat.v1.size(\n input, name=None, out_type=None\n )\n\nReturns a 0-D `Tensor` representing the number of elements in `input`\nof type `out_type`. Defaults to tf.int32.\n\n#### For example:\n\n t = tf.constant([[[1, 1, 1], [2, 2, 2]], [[3, 3, 3], [4, 4, 4]]])\n tf.size(t) # 12\n\n\u003cbr /\u003e\n\n\u003cbr /\u003e\n\n\u003cbr /\u003e\n\n| Args ---- ||\n|------------|----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|\n| `input` | A `Tensor` or `SparseTensor`. |\n| `name` | A name for the operation (optional). |\n| `out_type` | (Optional) The specified non-quantized numeric output type of the operation. Defaults to [`tf.int32`](../../../tf#int32). (Note: there is an experimental flag, `tf_shape_default_int64` that changes the default to [`tf.int64`](../../../tf#int64). This is an unsupported, experimental setting that causes known breakages.) |\n\n\u003cbr /\u003e\n\n\u003cbr /\u003e\n\n\u003cbr /\u003e\n\n\u003cbr /\u003e\n\n| Returns ------- ||\n|---|---|\n| A `Tensor` of type `out_type`. Defaults to [`tf.int32`](../../../tf#int32). ||\n\n\u003cbr /\u003e\n\n\u003cbr /\u003e\n\nnumpy compatibility\n-------------------\n\n\u003cbr /\u003e\n\nEquivalent to np.size()\n\n\u003cbr /\u003e\n\n\u003cbr /\u003e"]]