张量变形。

1.张量变形

所谓张量的变形，就是改变张量的形状。形状包括：张量的维度，不同维度方向上的长度。 一般情况下，变形前后，张量的总元素的个数和内容不变。

常见的变形操作如下：

• reshape：这是常用的变形的操作，可以直接进行维度、长度的变形。
• t：只适合与2维/2阶张量，它把行和列进行转置，对调。
• transpose：交换两个维度dim方向上的数据，而张量的维度不变，但不同维度方向上的长度发生了变化。
• squeeze：去除掉在某个dim方向上，向量长度为1的维度，实际上会导致降维。
• unblind：直接去除某个维度。
• unsqueeze：在某个dim方向上增加一个维度。
• flip：以某个维度方向为中心翻转向量。
• rot90：按照指定维度和旋转次数进行张量旋转。

import torch
print("===torch.reshape===")
a = torch.rand(2, 3)
print(a)
out = torch.reshape(a, (3, 2))
print(out)

print("===torch.t===")
print(torch.t(out))

print("===torch.transpose===")
print(out)
print(torch.transpose(out, 0, 1))

a = torch.rand(1, 2, 3)
print(a)
print(a.shape)
out = torch.transpose(a, 0, 1)
print(out)
print(out.shape)

print("===torch.squeeze===")
print(a)
print(a.shape)
out = torch.squeeze(a)
print(out)
print(out.shape)

print("===torch.unsqueeze===")
print(a)
print(a.shape)
out = torch.unsqueeze(a, -1)
print(out)
print(out.shape)

print("===torch.unbind===")
print(a)
print(a.shape)
out = torch.unbind(a, dim=2)
print(out)


print("===torch.flip===")
print(a)
print(torch.flip(a, dims=[2, 1]))

print("===torch.rot90===")
print(a)
print(a.shape)
out = torch.rot90(a, -1, dims=[0, 2])
print(out)
print(out.shape)