在 TensorFlow 中使用批次实现 zigzag flatten NxN 张量

【问题标题】：Implement zigzag flatten NxN tensor with batches in TensorFlow在 TensorFlow 中使用批次实现 zigzag flatten NxN 张量
【发布时间】：2021-07-10 06:44:55
【问题描述】：

问题可以在zigzag scanning 中描述。但是，我想知道是否有 TensorFlow 版本的实现，使用 TensorFlow 建议的 tf.tensor_scatter_nd_update 之类的东西。

BxNxN 张量，其中 B 代表 Batch。

【问题讨论】：

标签： tensorflow tensorflow2.0 tensorflow2.x

【解决方案1】：

我找到了使用 1x1 转换的解决方法。使用 numpy 生成一个常量置换 conv 内核（ tf 不支持 Eager tensor assignment... ），然后在应用tf.nn.conv2d 之前，将张量（BxNxN）重塑为 Bx1x1x（NxN）。最后做一些 reshape acrobat 来压平它。

【讨论】：