TensorFlow – 如何在N-D网格上广播参数进行评估

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

在使用TensorFlow的过程中，有些操作会自动广播参数，有时我们必须明确地广播参数。为了明确地广播参数，我们使用了Meshgrid方法。

使用的方法:

• meshgrid：该方法用于在N-D网格上广播参数的评估。它接受等级为1的张量，并将它们全部广播到相同的形状，并返回一个等级为N的张量列表。

例子1：在这种方法中，使用了默认的索引。

# importing the library
import tensorflow as tf
  
# Initializing Input
x = [1, 2, 3]
y = [4, 5, 6, 7]
  
# Printing the Input
print("x: ", x)
print("y: ", y)
  
# Broadcasting the Tensors
X, Y = tf.meshgrid(x, y)
  
# Printing the resulting Tensors
print("X: ", X)
print("Y: ", Y)

输出:

x:  [1, 2, 3]
y:  [4, 5, 6, 7]
X:  tf.Tensor(
[[1 2 3]
 [1 2 3]
 [1 2 3]
 [1 2 3]], shape=(4, 3), dtype=int32)
Y:  tf.Tensor(
[[4 4 4]
 [5 5 5]
 [6 6 6]
 [7 7 7]], shape=(4, 3), dtype=int32)

例子2：在这个例子中，索引被改成了’ij’。

# importing the library
import tensorflow as tf
  
# Initializing Input
x = [1, 2, 3]
y = [4, 5, 6, 7]
  
# Printing the Input
print("x: ", x)
print("y: ", y)
  
# Broadcasting the Tensors
X, Y = tf.meshgrid(x, y, indexing = 'ij')
  
# Printing the resulting Tensors
print("X: ", X)
print("Y: ", Y)

输出:

x:  [1, 2, 3]
y:  [4, 5, 6, 7]
X:  tf.Tensor(
[[1 1 1 1]
 [2 2 2 2]
 [3 3 3 3]], shape=(3, 4), dtype=int32)
Y:  tf.Tensor(
[[4 5 6 7]
 [4 5 6 7]
 [4 5 6 7]], shape=(3, 4), dtype=int32)