Pytorch——unsqueeze()函数理解

起到升维的作用，dim等于几表示在第几维度加一个维度

例子：

weights = torch.ones((2, 10)) * 0.1
values = torch.arange(20.0).reshape((2, 10))
a = weights.unsqueeze(1) # shape:(2,1,10) 在第1维度加一个维度，即增加的是第二个维度
b = values.unsqueeze(-1) # shape:(2,10,1) 在倒数第一个维度加一个维度，即增加的是倒数第一个维度
torch.bmm(a,b) # shape:(2,1,1)

结果：

tensor([[[ 4.5000]],
        [[14.5000]]])

参考资料：

1. python中torch.unsqueeze()函数理解