If I have a tensor like this
tensor([[0.3843, 0.3552]],
[[0.1743, 0.2439]],
[[0.3474, 0.1554]],
[[0.1325, 0.0778]],
[[0.2280, 0.1096]],
[[0.3842, 0.3997]],
[[0.1599, 0.0283]],
[[0.1876, 0.0815]]])
How could I get a tensor from him like that:
tensor([[0.3843]],
[[0.1743]],
[[0.3474]],
[[0.1325]],
[[0.2280]],
[[0.3842]],
[[0.1599]],
[[0.1876]]])