Tensorflow和cudnn卷积有不同的填充。但CPU / GPU卷积对用户来说是类似的。 tensorflow如何使用cudnn制作" tensoflow style"卷积?
答案 0 :(得分:0)
我建议你看一下tf.nn.convolution的实现(在tensorflow / python / ops / nn_ops.py中),看看它是如何从python中完成的,以及在tensorflow / core / kernels /中实现conv2d内核conv_ops.cc用于C ++位(调用cudnn)。