You have a tensor x = torch.tensor([1, 2, 3, 4]) with shape (4,). You perform the following operations: stacked_x = torch.stack([x, x, x], dim=0) stacked_y = torch.stack([x, x, x], dim=1) What are the shapes of stacked_x and stacked_y respectively?

Introduction to Deep Learning and Generative AI · Quiz 1 · 26 Oct 2025 · September 2025 term

Question 2: You have a tensor x = torch.tensor([1, 2, 3, 4]) with sha…

Question 2

+2 marksOne correct option

You have a tensor x = torch.tensor([1, 2, 3, 4]) with shape (4,). You perform the following operations:

python

1stacked_x = torch.stack([x, x, x], dim=0)
2stacked_y = torch.stack([x, x, x], dim=1)

What are the shapes of stacked_x and stacked_y respectively?

A
(3, 4) and (3, 4)
B
(3, 4) and (4, 3)
C
(4, 3) and (4, 3)
D
(4, 3) and (3, 4)

Show answer

Correct answer

B
(3, 4) and (4, 3)

Video solution

The video solution to this question is being recorded and will play here. Until then, the answer key is above.

See the whole paper Take it as a mock test

Question 2 of 20 in the IIT Madras BS Introduction to Deep Learning and Generative AI (Deep Learning and GenAI) Quiz 1 paper sat on 26 Oct 2025, in the September 2025 term (IIT M DIPLOMA AN EXAM QDD2 26 Oct 2025). It carries 2 marks.

Question 2: You have a tensor x = torch.tensor([1, 2, 3, 4]) with sha…

Question 2

Video solution

More questions from this paper