From 9aaa3a1bd7167a196bcb25a862c079550fd554e7 Mon Sep 17 00:00:00 2001 From: Vasilis Vryniotis Date: Tue, 15 Mar 2022 18:59:55 +0000 Subject: [PATCH] [FBcode->GH] JIT scriptable _max_value D34878986 --- torchvision/transforms/functional_tensor.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/torchvision/transforms/functional_tensor.py b/torchvision/transforms/functional_tensor.py index fba2ca1cad4..6bcd1ea85da 100644 --- a/torchvision/transforms/functional_tensor.py +++ b/torchvision/transforms/functional_tensor.py @@ -46,15 +46,15 @@ def get_image_num_channels(img: Tensor) -> int: def _max_value(dtype: torch.dtype) -> int: if dtype == torch.uint8: - return int(2 ** 8) - 1 + return 255 elif dtype == torch.int8: - return int(2 ** 7) - 1 + return 127 elif dtype == torch.int16: - return int(2 ** 15) - 1 + return 32767 elif dtype == torch.int32: - return int(2 ** 31) - 1 + return 2147483647 elif dtype == torch.int64: - return int(2 ** 63) - 1 + return 9223372036854775807 else: return 1