Commit aae0dc65 authored by Baiyu Zhao's avatar Baiyu Zhao Committed by Alykhan Tejani
Browse files

a small error in docstring of transforms.Normalize (#341)

docstring fix for Normalize: the std sequence should be (S1, ..., Sn).
parent c4b86b07
...@@ -94,7 +94,7 @@ class ToPILImage(object): ...@@ -94,7 +94,7 @@ class ToPILImage(object):
class Normalize(object): class Normalize(object):
"""Normalize an tensor image with mean and standard deviation. """Normalize an tensor image with mean and standard deviation.
Given mean: ``(M1,...,Mn)`` and std: ``(M1,..,Mn)`` for ``n`` channels, this transform Given mean: ``(M1,...,Mn)`` and std: ``(S1,..,Sn)`` for ``n`` channels, this transform
will normalize each channel of the input ``torch.*Tensor`` i.e. will normalize each channel of the input ``torch.*Tensor`` i.e.
``input[channel] = (input[channel] - mean[channel]) / std[channel]`` ``input[channel] = (input[channel] - mean[channel]) / std[channel]``
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment