tf.zeros(), tf.zeros_like(), tf.ones(),tf.ones_like()
因为tf.ones(),tf.ones_like()与tf.zeros(),tf.zeros_like()的用法相似,因此下面只演示前者的使用方法。
创建一个值为0的常量。
tf.zeros(shape, dtype=tf.float32, name=None):
shape:张量形状;
dtype:类型;
name:名称。
代码:
zeros_b = tf.zeros(shape=[2, 3], dtype=tf.int32) # 创建2x3矩阵,元素值均为0
根据输入张量创建一个值为0的张量,形状和输入张量相同。
tf.zeros_like(input_tensor, dtype=None, name=None, optimize=True):
input_tensor:张量;
dtype:类型;
name:名称;
optimize:优化。
代码:
zeros_like_c = tf.zeros_like(const_a)
查看生成数据
zeros_like_c.numpy()
输出:
array([[0., 0.],
[0., 0.]], dtype=float32)