Hi,
Yes, conv2d can only work with 4D inputs. You can use conv3d to work with 5D.
And we are definitely missing some error checking here to make sure the user is not providing wrong arguments. Could you open an issue on github about that please?