使用相同的值填充张量元素

下表列出了一些张量生成库函数,用于使用相同的值填充张量的所有元素:

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)

results matching ""

    No results matching ""