我正在尝试在GPU上的tensorflow中实现3D CVNN,但我的大多数变量都是复杂的64。当我的所有变量都是float32时,一切正常。
我已经对一些相关问题进行了分类,但我仍然坚持这个错误:
tensorflow.python.framework.errors.InvalidArgumentError: No OpKernel
was registered to support Op 'Conv3D' with these attrs. Registered kernels:
device='CPU'; T in [DT_DOUBLE]
device='CPU'; T in [DT_FLOAT]
device='GPU'; T in [DT_FLOAT]
[[Node: Conv3D = Conv3D[T=DT_COMPLEX64, padding="SAME",
strides=[1, 1, 1, 1, 1]](Reshape, Complex)]]
tensorflow是否对3D convnets有复杂的支持?或者它可能是我自己实现的错误?
答案 0 :(得分:0)
不,Tensorflow尚不支持复杂的3D卷积。我建议你在Tensorflow的github问题页面上打开一个功能请求。
更好的是,实现复杂的3D卷积支持并发送拉取请求: - )