function, mean, median, pytorch

mean() and median() in PyTorch

mean() can get the mean(average) values as shown below: *Memos: mean() can be called both from torch and a tensor. […]