TensorFlow 返回广播形状

tf.broadcast_dynamic_shape


broadcast_dynamic_shape (
shape_x ,
shape_y

定义在:tensorflow/python/ops/array_ops.py

参见指南:张量变换>张量形状的确定与改变

返回 shape_x 和 shape_y 之间的广播动态形状。

ARGS:

  • shape_x:一个秩为1的整数张量,表示 x 的形状。
  • shape_y:一个秩为1的整数张量,表示 y 的形状。

返回:

返回表示广播形状的秩为1的整数张量。

tf.broadcast_static_shape


broadcast_static_shape (
shape_x ,
shape_y

定义在:tensorflow/python/ops/array_ops.py

参见指南:张量变换>张量形状的确定与改变

返回 shape_x 和 shape_y 之间的广播静态形状。

ARGS:

  • shape_x: 一个 TensorShape
  • shape_y: 一个 TensorShape

返回:

表示广播形状的 TensorShape。

注意:

  • ValueError:当两个形状不能广播时。

发表评论

电子邮件地址不会被公开。 必填项已用*标注