torch.mean — PyTorch 1.10.1 documentation
pytorch.org › docs › stabletorch.mean(input, dim, keepdim=False, *, dtype=None, out=None) → Tensor. Returns the mean value of each row of the input tensor in the given dimension dim. If dim is a list of dimensions, reduce over all of them. If keepdim is True, the output tensor is of the same size as input except in the dimension (s) dim where it is of size 1.
torch.Tensor.mean — PyTorch 1.10.1 documentation
pytorch.org › generated › torchLearn about PyTorch’s features and capabilities. Community. Join the PyTorch developer community to contribute, learn, and get your questions answered. Developer Resources. Find resources and get questions answered. Forums. A place to discuss PyTorch code, issues, install, research. Models (Beta) Discover, publish, and reuse pre-trained models