share_memory

UNet.share_memory() torch.nn.modules.module.T

See torch.Tensor.share_memory_()