python - 如何在 Tensorflow 中复制 PyTorch 的 nn.functional.unfold 函数?
问题描述
我想用tensorflow重写pytorch的torch.nn.functional.unfold
功能:
#input x:[16, 1, 50, 36]
x = torch.nn.functional.unfold(x, kernel_size=(5, 36), stride=3)
#output x:[16, 180, 16]
我尝试使用该功能tf.extract_image_patches()
:
x = tf.extract_image_patches(x,ksizes=[1, 1,5, 98],strides=[1, 1, 3, 1], rates=[1, 1, 1, 1],padding='VALID')
输入x.shape
:[16,1,64,98]
我得到输出x.shape
:[16,1,20,490]
然后我重塑X
to [16,490,20]
,这是我所期望的。
但是当我提供数据时出现错误:
UnimplementedError (see above for traceback): Only support ksizes across space.
[[Node:hcn/ExtractImagePatches = ExtractImagePatches[T=DT_FLOAT, ksizes=[1, 1, 5, 98], padding="VALID", rates=[1, 1, 1, 1], strides=[1, 1, 3, 1], _device="/job:localhost/replica:0/task:0/device:GPU:0"](hcn/Reshape)]]
如何使用 tensorflow 重写 pytorchtorch.nn.functional.unfold
函数来更改X
?
解决方案
x = tf.reshape(x, [16, 50, 36, 1])
x = tf.extract_image_patches(x, ksizes=[1, 4, 98, 1], strides=[1, 4, 1, 1], rates=[1, 1, 1, 1], padding='VALID')
推荐阅读
- java - Java GUI 当我按下 JButton 时如何进行事件?
- gcc - 如何在 glibc 中使用 langinfo.h 功能而不破坏基于 FreeBSD 的系统中的编译?
- arrays - 计算C中堆栈中元素的数量
- hyperledger-fabric - 筏子订购服务可以设置同意者吗?
- javascript - 如何在javascript中的异步等待中返回一个承诺?
- laravel - Nova Laravel 单元测试期望返回 201,但返回 403
- angular - 在 Azure 服务器上访问平均应用程序的 URL 时出现问题
- c# - CS1002 ; 预期的。我有一个分号
- javascript - 如何使用 typescript 和 node 从依赖项中指定类型?
- ssl - 浏览器是否为 WebRTC 会话生成自签名证书?