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)