Python ``tensorflow.fill()`` 函数使用说明


发布日期 : 2021-10-15 13:16:51 UTC

访问量: 10 次浏览

Python – tensorflow.fill()

TensorFlow是谷歌设计的开源Python库,用于开发机器学习模型和深度学习神经网络。

fill() 用于生成一个具有标量值的张量。

语法:
tensorflow.fill( dims, value, name)

参数:

  • dims :它是dtype int32或int64的1-D序列,用非负整数代表产生的张量的形状。
  • value : 是指要填充的值。
  • name(可选): 它定义了操作的名称。

返回:
它返回一个形状为dim的张量。

抛出:

  • InvalidArgumentError :当dim包含负值时,会产生这个错误。
  • NotFoundError :当dim包含非整数值时,会产生这个错误。

示例 1:

# Importing the library
import tensorflow as tf

# Initializing the input
dim = [4, 5]
value = 5

# Printing the input
print('dim: ', dim)
print('value: ', value)

# Calculating result
res = tf.fill(dim, value)

# Printing the result
print('res: ', res)

输出:

dim: [4, 5]
value: 5
res: tf.Tensor(
[[5 5 5 5 5]
[5 5 5 5 5]
[5 5 5 5 5]
[5 5 5 5 5]], shape=(4, 5), dtype=int32)

示例 2:

# Importing the library
import tensorflow as tf

# Initializing the input
dim = [4, 2, 5]
value = 5

# Printing the input
print('dim: ', dim)
print('value: ', value)

# Calculating result
res = tf.fill(dim, value)

# Printing the result
print('res: ', res)

输出:

dim: [4, 2, 5]
value: 5
res: tf.Tensor(
[[[5 5 5 5 5]
[5 5 5 5 5]]

[[5 5 5 5 5]
[5 5 5 5 5]]

[[5 5 5 5 5]
[5 5 5 5 5]]

[[5 5 5 5 5]
[5 5 5 5 5]]], shape=(4, 2, 5), dtype=int32)