Fatal编程技术网
  • tensorflow/
  • 在TensorFlow中批量实现Z字形展平NxN张量

在TensorFlow中批量实现Z字形展平NxN张量

tensorflow

在TensorFlow中批量实现Z字形展平NxN张量,tensorflow,tensorflow2.0,tensorflow2.x,Tensorflow,Tensorflow2.0,Tensorflow2.x,问题可在中描述。但是，我想知道是否有TensorFlow版本的实现，它使用TensorFlow建议的tf.tensor\u scatter\u and\u update BxNxN张量，其中B表示批。我通过使用1x1 conv找到了一个解决方法。使用numpy生成一个常量置换conv内核（tf不支持急切张量赋值…），然后 在对其应用tf.nn.conv2d之前，将张量（BxNxN）重塑为Bx1x1x（NxN）。最后做一些重塑杂技，使其变平

问题可在中描述。但是，我想知道是否有TensorFlow版本的实现，它使用TensorFlow建议的
tf.tensor\u scatter\u and\u update

BxNxN张量，其中B表示批。

我通过使用1x1 conv找到了一个解决方法。使用numpy生成一个常量置换conv内核（tf不支持急切张量赋值…），然后 在对其应用
tf.nn.conv2d
之前，将张量（BxNxN）重塑为Bx1x1x（NxN）。最后做一些重塑杂技，使其变平