share_memory

UNet.share_memory() T

See torch.Tensor.share_memory_()