Tensor flattening

In this exercise, you need to flatten a tensor. Flattening a tensor means to convert it into a one-dimensional tensor.