numpy没有unsqueeze函数

最新推荐文章于 2024-03-13 13:27:03 发布

seven_不是赛文

最新推荐文章于 2024-03-13 13:27:03 发布

阅读量1.6k

点赞数

分类专栏： python实操文章标签： numpy python 机器学习

本文链接：https://blog.csdn.net/weixin_51015047/article/details/125976746

版权

python实操专栏收录该内容

62 篇文章 3 订阅

订阅专栏

unsqueeze：在指定数组中加入长度为1的维度。

import torch
arr = torch.Tensor(3, 5)
# print(type(arr), arr, arr.shape, sep='\n')  #torch.Size([3, 5])
arr_1 = arr.unsqueeze(0) #torch.Size([1, 3, 5]) ###‘头部加维度’
# print(type(arr_1), arr_1, arr_1.shape, sep='\n')
arr_2 = arr.unsqueeze(-1) #torch.Size([3, 5, 1])  ###‘尾部加维度’
# print(type(arr_2), arr_2, arr_2.shape, sep='\n')
arr_3 = arr.unsqueeze(1)#torch.Size([3, 1, 5])   ###‘中间加维度’
# print(type(arr_3), arr_3, arr_3.shape, sep='\n')
arr_4 = arr.unsqueeze(2)        # 数字再大就报错了  #torch.Size([3, 5, 1]) 和“##（-1）一样
print(type(arr_4), arr_4, arr_4.shape, sep='\n')