Fixed the problem that when torch.add(1, vary_large_tensor) and torch.div(2, vary_large_tensor), a temporary tensor of the same size as vary_large_tensor is allocated and then the value is filled in, which reduces the waste of device memory and unnecessary memory access overhead.
Fixed the problem that when torch.add(1, vary_large_tensor) and torch.div(2, vary_large_tensor), a temporary tensor of the same size as vary_large_tensor is allocated and then the value is filled in, which reduces the waste of device memory and unnecessary memory access overhead.