当前位置: 首页>>代码示例 >>用法及示例精选 >>正文

Python tf.ones用法及代码示例


创建一个所有元素都设置为一 (1) 的张量。

用法

tf.ones(
    shape, dtype=tf.dtypes.float32, name=None
)

参数

  • shape 整数的 list、整数的 tuple 或 int32 类型的一维 Tensor 。
  • dtype 生成的 Tensor 中元素的可选 DType。默认为 tf.float32 。
  • name 可选字符串。操作的名称。

返回

  • 一个Tensor，所有元素都设置为一 (1)。

另见tf.ones_like、tf.zeros、tf.fill、tf.eye。

此操作返回一个类型为 dtype 的张量，形状为 shape，所有元素都设置为 1。

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)>

相关用法


注：本文由纯净天空筛选整理自tensorflow.org大神的英文原创作品 tf.ones。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。