def eval_partial(x_ma_flat): x_ma = x_ma_flat.reshape(-1,D) return tps.tps_eval(x_ma, lin_ag, trans_g, w_ng, pts0)
def transform_points(self, x_ma): y_ng = tps.tps_eval(x_ma, self.lin_ag, self.trans_g, self.w_ng, self.x_na) return y_ng
def eval_partial(x_ma_flat): x_ma = x_ma_flat.reshape(-1, 3) return tps.tps_eval(x_ma, lin_ag, trans_g, w_ng, pts0)