Mean.py « nn « legacy « torch - platform/upstream/pytorch - Domain: Machine Learning / ML Framework; Licenses: BSD-3-Clause;

index : platform/upstream/pytorch

Domain: Machine Learning / ML Framework; Licenses: BSD-3-Clause;

Geunsik Lim <geunsik.lim@samsung.com>, HyoungJoo Ahn <hello.ahn@samsung.com>, Jijoong Moon <jijoong.moon@samsung.com>, MyungJoo Ham <myungjoo.ham@samsung.com>, Parichay Kapoor <pk.kapoor@samsung.com>

summary refs log tree commit diff

path: root/torch/legacy/nn/Mean.py

blob: 905d91dbbfbddffab0af8259b55bd3a846169983 (plain)

import torch
from .Sum import Sum

"""

This file is still here because of backward compatibility.

Please use instead "nn.Sum(dimension, nInputDims, sizeAverage)"

"""

class Mean(Sum):

    def __init__(self, dimension):
        super(Mean, self).__init__(dimension, True)