View source on GitHub
|
Creates a tensor with all elements set to zero.
tf.zeros(
shape,
dtype=tf.dtypes.float32,
name=None,
layout=None
)
See also tf.zeros_like, tf.ones, tf.fill, tf.eye.
This operation returns a tensor of type dtype with shape shape and
all elements set to zero.
tf.zeros([3, 4], tf.int32)<tf.Tensor: shape=(3, 4), dtype=int32, numpy=array([[0, 0, 0, 0],[0, 0, 0, 0],[0, 0, 0, 0]], dtype=int32)>
Args | |
|---|---|
shape
|
A list of integers, a tuple of integers, or a 1-D Tensor of
type int32.
|
dtype
|
The DType of an element in the resulting Tensor.
|
name
|
Optional string. A name for the operation. |
layout
|
Optional, tf.experimental.dtensor.Layout. If provided, the result
is a DTensor with the
provided layout.
|
Returns | |
|---|---|
A Tensor with all elements set to zero.
|
View source on GitHub