pymc3/pymc3/gp/mean.py seems to assume the same shape for X and Y in Class Zero and Constant #2078
Description
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