标签： tensorflow 3d neural-network convolution
我正在构建一个3D卷积NN，它具有一个张量的形状（100,50,50,50,1）。我使用tf.conv3d，过滤器大小为（4,4,4）。当我只使用一个样本（1,50,50,50,1）而不是批量训练时，代码工作。使用批量训练计算conv3d需要花费很多时间。有人有任何想法解决这个问题吗？谢谢！