torch.unbind()拆分张量练习过程
import torch# 创建一个 3D 张量,形状为 (5, 4, 3)
a = torch.tensor([[[1, 2, 3], [4, 5, 6], [7, 8, 9], [10, 11, 12]],[[13, 14, 15], [16, 17, 18], [19, 20, 21], [22, 23, 24]],[[25, 26, 27], [28, 29, 30], [31, 32, 33], [34, 35, 36]],[[37, 38, 39], [40, 41, 42], [43, 44, 45], [46, 47, 48]],[[49, 50, 51], [52, 53, 54], [55, 56, 57], [58, 59, 60]]])# 使用 unbind 在 dim=0 上拆分,按照(5,4,3)->得到5个4*3的张量
n = a.unbind(dim=0)# 输出每个拆分后的张量
for i, tensor in enumerate(n):print(f"Tensor {i}: {tensor}")
# 使用 unbind 在 dim=0 上拆分,按照(5,4,3)->得到5个4*3的张量
import torch# 创建一个 3D 张量,形状为 (5, 4, 3)
a = torch.tensor([[[1, 2, 3], [4, 5, 6], [7, 8, 9], [10, 11, 12]],[[13, 14, 15], [16, 17, 18], [19, 20, 21], [22, 23, 24]],[[25, 26, 27], [28, 29, 30], [31, 32, 33], [34, 35, 36]],[[37, 38, 39], [40, 41, 42], [43, 44, 45], [46, 47, 48]],[[49, 50, 51], [52, 53, 54], [55, 56, 57], [58, 59, 60]]])# 使用 unbind 在 dim=1 上拆分,按照(5,4,3)->得到4个5*3的张量
n = a.unbind(dim=1)# 输出每个拆分后的张量
for i, tensor in enumerate(n):print(f"Tensor {i}: {tensor}")
# 使用 unbind 在 dim=1 上拆分,按照(5,4,3)->得到4个5*3的张量
import torch# 创建一个 3D 张量,形状为 (5, 4, 3)
a = torch.tensor([[[1, 2, 3], [4, 5, 6], [7, 8, 9], [10, 11, 12]],[[13, 14, 15], [16, 17, 18], [19, 20, 21], [22, 23, 24]],[[25, 26, 27], [28, 29, 30], [31, 32, 33], [34, 35, 36]],[[37, 38, 39], [40, 41, 42], [43, 44, 45], [46, 47, 48]],[[49, 50, 51], [52, 53, 54], [55, 56, 57], [58, 59, 60]]])# 使用 unbind 在 dim=2 上拆分,按照(5,4,3)->得到3个5*4的张量
n = a.unbind(dim=2)# 输出每个拆分后的张量
for i, tensor in enumerate(n):print(f"Tensor {i}: {tensor}")
# 使用 unbind 在 dim=2 上拆分,按照(5,4,3)->得到3个5*4的张量
初始张量如果维度是2D,则dim=0 or 1 ,另外《pytorch速查手册》P45认为,torch.t() 要求输入是一个2维张量,并转置维度0和1 ,我认为转置后还是2.,在编程学习中,书是参考,以实际操作为准并修订书中的内容。