change tensor to as_tensor (#810)

1fa2f866 · ekka · Francisco Massa · 6eac421c · 1fa2f866
Commit 1fa2f866 authored Mar 24, 2019 by ekka Committed by Francisco Massa Mar 24, 2019
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

torchvision/transforms/functional.py torchvision/transforms/functional.py +2 -2

No files found.
--- a/torchvision/transforms/functional.py
+++ b/torchvision/transforms/functional.py
@@ -203,8 +203,8 @@ def normalize(tensor, mean, std, inplace=False):
    if not inplace:
        tensor = tensor.clone()

-    mean = torch.tensor(mean, dtype=torch.float32, device=tensor.device)
-    std = torch.tensor(std, dtype=torch.float32, device=tensor.device)
+    mean = torch.as_tensor(mean, dtype=torch.float32, device=tensor.device)
+    std = torch.as_tensor(std, dtype=torch.float32, device=tensor.device)
    tensor.sub_(mean[:, None, None]).div_(std[:, None, None])
    return tensor