微信公众号搜"智元新知"关注
微信扫一扫可直接关注哦!

在 TensorFlow 中批量实现锯齿形展平 NxN 张量

如何解决在 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 举报,一经查实,本站将立刻删除。