Create and Inspect a Tensor

Easypytorch

Write a function make_tensor() that creates and returns a 2x3 PyTorch tensor of dtype float32 with the values:

$\begin{bmatrix} 1 & 2 & 3 \\ 4 & 5 & 6 \end{bmatrix}$

The returned tensor must have shape (2, 3) and dtype torch.float32.

Example:

Input:

t = make_tensor()
print(t)
print(tuple(t.shape))
print(t.dtype)

Output:

tensor([[1., 2., 3.],
        [4., 5., 6.]])
(2, 3)
torch.float32

Reasoning:

make_tensor builds a 2x3 float32 tensor from the nested list [[1, 2, 3], [4, 5, 6]]. Printing shows float formatting, shape (2, 3), and dtype torch.float32.

Learn About topic

Contributors:

No contributors for this problem.

Contribute