使用相同的值填充张量元素
下表列出了一些张量生成库函数,用于使用相同的值填充张量的所有元素:
zeros(
shape,
dtype=tf.float32,
name=None
)
创建所提供形状的张量,所有元素都设置为零
zeros_like(
tensor,
dtype=None,
name=None,
optimize=True
)
创建与参数形状相同的张量,所有元素都设置为零
ones(
shape,
dtype=tf.float32,
name=None
)
创建所提供形状的张量,所有元素都设置为 1
ones_like(
tensor,
dtype=None,
name=None,
optimize=True
)
创建与参数形状相同的张量,所有元素都设置为 1
fill(
dims,
value,
name=None
)
创建一个形状的张量作为dims
参数,,所有元素都设置为value
;例如,a = tf.fill([100],0)