relu.py 文件源码

def __init__(self,shape,z0rep_axes=(0,), z1rep_axes=(0,), map_est=False):
        Estim.__init__(self)
        self.shape = shape
        ndim = len(shape)
        if z0rep_axes == 'all':
            z0rep_axes = tuple(range(ndim))
        if z1rep_axes == 'all':
            z1rep_axes = tuple(range(ndim))            
        self.z0rep_axes = z0rep_axes
        self.z1rep_axes = z1rep_axes
        self.cost_avail = True
        self.map_est = map_est

        # Initial variances
        self.zvar0_init= np.Inf
        self.zvar1_init= np.Inf