Premium problem44. Concatenate Tensors

Easy Locked

Concatenate the two 1-D tensors a and b end-to-end into a single 1-D tensor.

Input

a = tensor([1, 2, 3])
b = tensor([4, 5])

Output

tensor([1, 2, 3, 4, 5])

Premium problem

This one's part of Premium. Unlock the full PyTorch track plus every other premium problem on the site.

Implement solve(...)