Method expect

code/deep/CSG/distr/base.py:84–92

(self, fn, conds: edic=edic(), n_mc: int=10, repar: bool=True, reducefn = tc.mean)

        raise NotImplementedError

    def expect(self, fn, conds: edic=edic(), n_mc: int=10, repar: bool=True, reducefn = tc.mean) -> tc.Tensor:
        # [shape_bat] -> [shape_bat]
        if n_mc == 0:
            vals = self.mean(conds, 0, repar)
            return fn(conds|vals)
        elif n_mc > 0:
            vals = self.draw(tc.Size((n_mc,)), conds, repar)
            return reducefn(fn( edicify(conds)[0].expand_front((n_mc,)) | vals ), dim=0)
        else: raise ValueError(f"For {self}, negative `n_mc` {n_mc} encountered")

    def rdraw(self, shape_mc: tc.Size=tc.Size(), conds: edic=edic()) -> edic: # vals
        # shape_mc, [shape_bat, shape_cond] -> [shape_mc, shape_bat, shape_var]