pymc3/pymc3/gp/mean.py seems to assume the same shape for X and Y in Class Zero and Constant

[zero-you]

This causes an exception when I try to create a GP with 6-D inputs and 1-D output, e.g., X.shape=[100,6] and Y.shape=[100,1]

In pymc3/gp/mean.py, the return values of Zero and Constant have the same shape as X:

class Zero(Mean):
def call(self, X):
return tt.zeros(X.shape, dtype='float32')

class Constant(Mean):
def init(self, c=0):
Mean.init(self)
self.c = c
def call(self, X):
return tt.ones(X.shape) * self.c

[bwengals]

Good catch, this is definitely a bug. Are you planning on submitting a PR? If not, I am happy to.

[zero-you]

@bwengals Please go ahead. I am not sure what would be the best way to fix it - Maybe allow users to pass Y shape as an optional input argument?

[junpenglao]

Should be fixed now.