NumPy bridge (from_numpy, numpy) in PyTorch - ML Experiment: Train & Evaluate

Experiment - NumPy bridge (from_numpy, numpy)

Problem:You want to efficiently convert data between NumPy arrays and PyTorch tensors to use PyTorch models with existing NumPy data.

Current Metrics:Currently, you manually convert data using separate steps, which is slow and error-prone.

Issue:Manual conversions cause extra memory usage and slow down data processing.

Your Task

Use PyTorch's from_numpy and numpy() functions to convert data back and forth efficiently without extra memory copies.

You must use PyTorch and NumPy only.

Do not create new arrays or tensors unnecessarily.

Ensure changes in one reflect in the other.

Hint 1

Hint 2

Hint 3

Solution

PyTorch

import numpy as np
import torch

# Create a NumPy array
np_array = np.array([1, 2, 3, 4], dtype=np.float32)

# Convert NumPy array to PyTorch tensor (shares memory)
tensor = torch.from_numpy(np_array)

# Modify the tensor
tensor[0] = 10

# Check if NumPy array changed
print(f"NumPy array after tensor change: {np_array}")  # Should show [10.  2.  3.  4.]

# Convert tensor back to NumPy array (shares memory)
np_array2 = tensor.numpy()

# Modify the new NumPy array
np_array2[1] = 20

# Check if tensor changed
print(f"Tensor after NumPy array change: {tensor}")  # Should show tensor([10., 20.,  3.,  4.])

Used torch.from_numpy() to create tensor sharing memory with NumPy array.

Used tensor.numpy() to convert tensor back to NumPy array sharing memory.

Demonstrated that modifying one changes the other, confirming no extra copies.

Results Interpretation

Before: Manual conversions created copies, causing extra memory use and slower data handling.

After: Using from_numpy and numpy() shares memory between NumPy arrays and tensors, making conversions fast and memory efficient.

Sharing memory between NumPy arrays and PyTorch tensors avoids unnecessary data copying, improving performance and reducing memory use.

Bonus Experiment

Try modifying a tensor created from a NumPy array with requires_grad=True and observe how gradients behave.

💡 Hint

Remember that tensors sharing memory with NumPy arrays cannot have requires_grad=True directly; consider cloning or detaching.

Practice

1. What does torch.from_numpy() do in PyTorch?

easy

A. Converts a PyTorch tensor into a NumPy array

B. Creates a new PyTorch tensor with random values

C. Converts a NumPy array into a PyTorch tensor sharing the same memory

D. Converts a Python list into a NumPy array

Solution

1. Step 1: Understand the function purpose

   torch.from_numpy() is designed to take a NumPy array and create a PyTorch tensor from it.

2. Step 2: Memory sharing behavior

   The created tensor shares the same memory with the original NumPy array, so changes in one affect the other.

3. Final Answer:

   Converts a NumPy array into a PyTorch tensor sharing the same memory -> Option C

4. Quick Check:

   from_numpy() converts NumPy array to tensor [OK]

Hint: Remember from_numpy links tensor to NumPy array memory [OK]

Common Mistakes:

• Confusing from_numpy with tensor.numpy()
• Thinking it copies data instead of sharing memory
• Assuming it creates a tensor from a Python list

2. Which of the following is the correct syntax to convert a PyTorch tensor t back to a NumPy array?

easy

A. t.numpy()

B. t.to_numpy()

C. torch.from_numpy(t)

D. numpy(t)

Solution

1. Step 1: Identify the method to convert tensor to NumPy

   PyTorch tensors have a method called numpy() that returns a NumPy array sharing the same data.

2. Step 2: Check syntax correctness

   The correct syntax is calling t.numpy() on the tensor object.

3. Final Answer:

   t.numpy() -> Option A

4. Quick Check:

   tensor.numpy() converts tensor to NumPy array [OK]

Hint: Use tensor.numpy() to get NumPy array from tensor [OK]

Common Mistakes:

• Using numpy(t) which is invalid
• Confusing from_numpy with numpy()
• Using a non-existent method like to_numpy()

3. What will be the output of the following code?

import torch
import numpy as np
np_array = np.array([1, 2, 3])
tensor = torch.from_numpy(np_array)
np_array[0] = 10
print(tensor)

medium

A. Error because tensor is immutable

B. tensor([1, 2, 3])

C. tensor([0, 2, 3])

D. tensor([10, 2, 3])

Solution

1. Step 1: Understand memory sharing between NumPy and tensor

   The tensor created by torch.from_numpy() shares memory with the NumPy array, so changes in the array affect the tensor.

2. Step 2: Analyze the code change

   Changing np_array[0] to 10 updates the shared memory, so the tensor reflects this change.

3. Final Answer:

   tensor([10, 2, 3]) -> Option D

4. Quick Check:

   Shared memory means tensor updates with NumPy changes [OK]

Hint: Changing NumPy array updates tensor from from_numpy() [OK]

Common Mistakes:

• Assuming tensor is independent and unchanged
• Thinking tensor is immutable and throws error
• Confusing tensor.numpy() with from_numpy()

4. Identify the error in this code snippet:

import torch
import numpy as np
np_array = np.array([1, 2, 3])
tensor = torch.from_numpy(np_array)
tensor += 1
print(np_array)

medium

A. No error; np_array will print [2, 3, 4]

B. RuntimeError because tensor is not writable

C. SyntaxError due to invalid operator

D. np_array remains unchanged as tensor is a copy

Solution

1. Step 1: Understand memory sharing

   Tensors from torch.from_numpy() share memory with NumPy arrays, so in-place operations like += modify both.

2. Step 2: Analyze the operation

   tensor += 1 performs in-place addition on the shared memory. No error occurs, and np_array becomes [2, 3, 4].

3. Final Answer:

   No error; np_array will print [2, 3, 4] -> Option A

4. Quick Check:

   In-place ops on from_numpy tensors modify NumPy array [OK]

Hint: In-place ops on from_numpy tensors modify NumPy array [OK]

Common Mistakes:

• Thinking it causes RuntimeError because tensor not writable
• Assuming tensor is independent copy
• Confusing with syntax error

5. You have a NumPy array arr = np.array([1, 2, 3]). You want to create a PyTorch tensor from it, modify the tensor without affecting the original array, and then convert the tensor back to a NumPy array. Which code snippet achieves this correctly?

hard

A. tensor = torch.from_numpy(arr)
tensor[0] = 10
new_arr = tensor.numpy()

B. tensor = torch.tensor(arr)
tensor[0] = 10
new_arr = tensor.numpy()

C. tensor = torch.from_numpy(arr)
tensor.clone()
tensor[0] = 10
new_arr = tensor.numpy()

D. tensor = torch.tensor(arr)
arr[0] = 10
new_arr = tensor.numpy()

Solution

1. Step 1: Understand memory sharing and cloning

   torch.from_numpy() shares memory with the NumPy array, so modifying the tensor changes the array. Using torch.tensor() creates a copy, so changes don't affect the original.

2. Step 2: Check each option

   tensor = torch.from_numpy(arr) tensor[0] = 10 new_arr = tensor.numpy() shares memory, so modifies arr.
   tensor = torch.from_numpy(arr) tensor.clone() tensor[0] = 10 new_arr = tensor.numpy() calls clone() without assigning back to tensor, so tensor remains shared and modifies arr.
   tensor = torch.tensor(arr) tensor[0] = 10 new_arr = tensor.numpy() creates an independent copy, safe.
   tensor = torch.tensor(arr) arr[0] = 10 new_arr = tensor.numpy() modifies arr directly.

3. Final Answer:

   tensor = torch.tensor(arr) tensor[0] = 10 new_arr = tensor.numpy() -> Option B

4. Quick Check:

   Use torch.tensor() to avoid shared memory [OK]

Hint: Use torch.tensor() to copy and avoid shared memory [OK]

Common Mistakes:

• Using from_numpy and modifying tensor affects original array
• Calling clone() without reassigning the tensor
• Modifying original array instead of tensor