Creates a tensor with all elements set to one (1).
tfp.experimental.distributions.marginal_fns.ps.ones(
shape, dtype=dtypes.float32, name=None, layout=None
)
See also tf.ones_like
, tf.zeros
, tf.fill
, tf.eye
.
This operation returns a tensor of type dtype
with shape shape
and
all elements set to one.
tf.ones([3, 4], tf.int32)
<tf.Tensor: shape=(3, 4), dtype=int32, numpy=
array([[1, 1, 1, 1],
[1, 1, 1, 1],
[1, 1, 1, 1]], dtype=int32)>
Args | |
---|---|
shape
|
A list of integers, a tuple of integers, or a 1-D Tensor of
type int32 .
|
dtype
|
Optional DType of an element in the resulting Tensor . Default is
tf.float32 .
|
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 one (1).
|