pytorch torch.unsqueeze()
0
-
Pytorch中torch.unsqueeze()与torch.squeeze()函数详细解析
目录一. torch.squeeze()函数解析1. 官网链接2. torch.squeeze()函数解析3. 代码举例3.1 输入size=(2, 1, 2, 1, 2)的张量3.2 把x中维度大小为1的所有维都已删除3.3 把x中第一维删除,但是第一维大小为2,不为1,因此[详细]
2023-02-15 09:24 分类:开发