import numpy as np


def find_mean_std(dataset):
    for each in enumerate(dataset):
        print(each)

    temp = dataset.reshape(dataset.shape[-1], -1)

    print(temp.shape)

    return np.array(temp.mean(axis=1)) / 255.0, np.array(temp.std(axis=1)) / 255.0
