cntk_backend.py 275 Bytes
Newer Older
maming's avatar
maming committed
1
2
3
4
5
6
7
from keras.backend import cntk_backend as KCN


def moments(x, axes, shift=None, keep_dims=False):
    ''' Calculates and returns the mean and variance of the input '''
    mean, variant = KCN._moments(x, axes=axes, shift=shift, keep_dims=keep_dims)
    return mean, variant