From be8cb1080f8565652a7f76d3fae4a01e009abaa2 Mon Sep 17 00:00:00 2001 From: Nic Ma Date: Thu, 16 Sep 2021 10:29:33 +0800 Subject: [PATCH] [DLMED] fix float64 bug Signed-off-by: Nic Ma --- monai/transforms/spatial/array.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/monai/transforms/spatial/array.py b/monai/transforms/spatial/array.py index c49f4e6479..cb80c2036b 100644 --- a/monai/transforms/spatial/array.py +++ b/monai/transforms/spatial/array.py @@ -526,7 +526,7 @@ def __call__( align_corners=self.align_corners if align_corners is None else align_corners, reverse_indexing=True, ) - output: torch.Tensor = xform(img_t.unsqueeze(0), transform_t, spatial_size=output_shape).squeeze(0) + output: torch.Tensor = xform(img_t.unsqueeze(0), transform_t, spatial_size=output_shape).float().squeeze(0) self._rotation_matrix = transform out: NdarrayOrTensor out, *_ = convert_to_dst_type(output, dst=img, dtype=output.dtype)