| #------------------------------------------------------------- |
| # |
| # Licensed to the Apache Software Foundation (ASF) under one |
| # or more contributor license agreements. See the NOTICE file |
| # distributed with this work for additional information |
| # regarding copyright ownership. The ASF licenses this file |
| # to you under the Apache License, Version 2.0 (the |
| # "License"); you may not use this file except in compliance |
| # with the License. You may obtain a copy of the License at |
| # |
| # http://www.apache.org/licenses/LICENSE-2.0 |
| # |
| # Unless required by applicable law or agreed to in writing, |
| # software distributed under the License is distributed on an |
| # "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY |
| # KIND, either express or implied. See the License for the |
| # specific language governing permissions and limitations |
| # under the License. |
| # |
| #------------------------------------------------------------- |
| |
| N = 2 |
| C = 2 |
| Hin = 3 |
| Win = 4 |
| X = matrix(rbind(seq(1,20),seq(1,20),seq(1,8)), rows=2, cols=24) |
| gamma = matrix("1 2", rows=2, cols=1) |
| beta = matrix("0 1", rows=2, cols=1) |
| ema_mean = matrix("4 5", rows=2, cols=1) |
| ema_var = matrix("2 3", rows=2, cols=1) |
| mu = 0.95 |
| epsilon = 1e-4 |
| |
| subgrp_means = matrix(colMeans(X), rows=C, cols=Hin*Win) |
| subgrp_vars = matrix(colVars(X) * ((N-1)/N), rows=C, cols=Hin*Win) |
| mean = rowMeans(subgrp_means) # shape (C, 1) |
| var = rowMeans(subgrp_vars) + rowVars(subgrp_means)*(((Hin*Win)-1)/(Hin*Win)) |
| ema_mean_upd = mu*ema_mean + (1-mu)*mean |
| ema_var_upd = mu*ema_var + (1-mu)*var |
| |
| R = cbind(mean, var, ema_mean_upd, ema_var_upd) |
| |
| write(R,$1) |
