Skip to content

Commit c095f92

Browse files
committed
test new_empty
Signed-off-by: Wenqi Li <wenqil@nvidia.com>
1 parent a438179 commit c095f92

File tree

1 file changed

+11
-0
lines changed

1 file changed

+11
-0
lines changed

monai/data/meta_tensor.py

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -232,3 +232,14 @@ def affine(self) -> torch.Tensor:
232232
def affine(self, d: torch.Tensor) -> None:
233233
"""Set the affine."""
234234
self.meta["affine"] = d
235+
236+
def new_empty(self, size, dtype=None, device=None, requires_grad=False):
237+
"""
238+
must be defined for deepcopy to work
239+
240+
See:
241+
- https://pytorch.org/docs/stable/generated/torch.Tensor.new_empty.html#torch-tensor-new-empty
242+
"""
243+
return type(self)(
244+
self.as_tensor().new_empty(size=size, dtype=dtype, device=device, requires_grad=requires_grad)
245+
)

0 commit comments

Comments
 (0)