如何解决在 TensorFlow 中批量实现锯齿形展平 NxN 张量
问题可以用zigzag scanning来描述。但是,我想知道是否有使用 TensorFlow 建议的 tf.tensor_scatter_nd_update
之类的实现的 TensorFlow 版本。
BxNxN 张量,其中 B 代表批次。
解决方法
我找到了一种使用 1x1 转换的解决方法。使用 numpy 生成一个常数置换卷积核(tf 不支持 Eager tensor assignment...),然后
在对其应用 tf.nn.conv2d
之前将张量(BxNxN)重塑为 Bx1x1x(NxN)。最后做一些reshape acrobat 把它弄平。
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。