是否支持在张量流中具有复杂变量的3D卷积内核?

时间:2016-10-24 22:53:46

标签: tensorflow

我正在尝试在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有复杂的支持?或者它可能是我自己实现的错误?

1 个答案:

答案 0 :(得分:0)

不,Tensorflow尚不支持复杂的3D卷积。我建议你在Tensorflow的github问题页面上打开一个功能请求。

更好的是,实现复杂的3D卷积支持并发送拉取请求: - )