TensorFlow函数:tf.ones_like
2018-11-07 17:39 更新
tf.ones_like 函数
ones_like(
tensor,
dtype=None,
name=None,
optimize=True
)
定义在:tensorflow/python/ops/array_ops.py.
请参阅指南:生成常量,序列和随机值>常量值张量
创建一个将所有元素都设置为1的张量.
给定一个张量(tensor),此操作返回与所有元素设置为1的张量相同的类型和形状的张量.可选地,可以为返回的张量指定一个新的类型(dtype).
例如:
# 'tensor' is [[1, 2, 3], [4, 5, 6]]
tf.ones_like(tensor) ==> [[1, 1, 1], [1, 1, 1]]
参数:
- tensor:张量.
- dtype:返回的 Tensor 类型.必须是:float32,float64,int8,int16,int32,int64,uint8,complex64,complex128或bool.
- name:操作的名称(可选).
- optimize:如果为 true,则尝试静态确定 "张量" 的形状并将其编码为常量.
返回值:
该函数将返回所有元素设置为1的张量.