简单粗暴,看一下两个函数的用法:
1.unsqueeze():
从tensor(torch)或者数组中,在指定维度上添加维度为1的维度(axis=0, axis=1, axis=2, …)
以pytorch为例,首先初始化一个tensor,size为(2,3).
在tensor的第一维unsqueeze(0)
中插入维度为1的维度,现在tensor的size为(1,2,3)。
2.squeeze():
从tensor(torch)或者数组中,在指定维度上删除维度为1的维度(axis=0, axis=1, axis=2, …)
Note:
对于Tensor操作的几个函数,包括.reshape()
, .view()
, .permute()
。涉及到维度的变化,主要是用来进行矩阵乘法运算,一定要注意维度!!!