pytorch基础知识五【维度变换】

1. 改变形状view & reshape

(1).reshape(dim,dim)和.view(dim,dim) 改变张量的形状,但不改变其数据的总量。
(2)这两种方法虽然不改变张量的总量,但改变形状后会丢失原来的维度信息,无法复原。

a = torch.rand(2,3,4)
print(a)
b = a.reshape(6,4)
print(b)
c = a.view(4,6)
print(c)

执行结果:
	tensor([[[0.7205, 0.5070, 0.1591, 0.4621],
         [0.7140, 0.9463, 0.5383, 0.3105],
         [0.2181, 0.7825, 0.2277, 0.7061]],

        [[0.7025, 0.4321, 0.7696, 0.9744],
         [0.4337, 0.6441, 0.8214, 0.7411],
         [0.1081, 0.3400, 0.5136, 0.1967]]])
	tensor([[0.7205, 0.5070, 0.1591, 0.4621],
	        [0.7140, 0.9463, 0.5383, 0.3105],
	        [0.2181, 0.7825, 0.2277, 0.7061],
	        [0.7025, 0.4321, 0.7696, 0.9744],
	        [0.4337, 0.6441, 0.8214, 0.7411],
	        [0.1081, 0.3400, 0.5136, 0.1967]])
	tensor([[0.7205, 0.5070, 0.1591, 0.4621, 0.7140, 0.9463],
	        [0.5383, 0.3105, 0.2181, 0.7825, 0.2277, 0.7061],
	        [0.7025, 0.4321, 0.7696, 0.9744, 0.4337, 0.6441],
	        [0.8214, 0.7411, 0.1081, 0.3400, 0.5136, 0.1967]])


在这里插入图片描述

2. 增删维度squeeze & unsqueeze

2.1 增加维度unsqueeze()

在这里插入图片描述
在这里插入图片描述

在这里插入图片描述
举例:
在这里插入图片描述
在这里插入图片描述

2.2 删减维度squeeze()

(1) squeeze(dim) 只能删减数量为1的维度;
(2) squeeze() 不传参数表示把能删减的维度都删减;
(3) 遇到不能删减维度的张量,返回原张量。

在这里插入图片描述

3. 维度形状扩展expand & repeat

(1) expand() :broadcasting  扩展维度形状不增加数据;节约内存,推进使用
(2) repeat() :memory copied  扩展维度形状增加数据。

3.1 expand()

(1)expand() 扩展维度形状(shape)的时候,原来的形状必须是1。扩展维度形状时输入-1,扩展后形状保持不变。
(2)expand() 输入的参数是扩展后的结果,也就是新的shape
在这里插入图片描述

3.2 repeat()

(1)repeat() 输入的参数是张量每个维度形状扩展(或拷贝)的倍数,是乘法关系。
在这里插入图片描述

4. 二位矩阵(张量)转置操作.t()

.t() 转置操作只能用于二位张量。

在这里插入图片描述

5. 维度交换(两两交换)transpose()

注意:数据的内存连续与数据结构的破坏。
在这里插入图片描述

6. 维度交换(任意交换)permute()

在这里插入图片描述

  • 0
    点赞
  • 12
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
PyTorch中,可以使用多种方法对张量的维度进行变换。其中常用的方法有view()和squeeze()。view()方法可以用来改变张量的形状,而squeeze()方法可以去除维度中值为1的尺寸。举例如下: 1. 使用view()方法进行维度变换。view()方法会返回一个改变了形状的新张量,但张量中的元素数量必须保持不变。可以使用该方法实现维度的展平、增加或减少维度等操作。 2. 使用squeeze()方法进行维度变换。squeeze()方法可以去除张量中维度中值为1的尺寸,并返回一个新张量。可以指定具体的维度进行去除,也可以不指定维度,即默认去除所有为1的维度。 例如,假设有一个张量a的形状为(2, 1, 2, 1, 3),使用squeeze()方法可以去除其中值为1的维度,得到一个形状为(2, 2, 3)的新张量。 在PyTorch中,还可以使用unsqueeze()方法对张量进行维度扩展。unsqueeze()方法会在指定的维度上增加一个尺寸为1的维度,并返回一个新张量。 总结起来,PyTorch中的维度变换包括view()、squeeze()和unsqueeze()等方法,可以根据具体需求选择合适的方法进行张量的维度变换。<span class="em">1</span><span class="em">2</span><span class="em">3</span> #### 引用[.reference_title] - *1* [PyTorch中Tensor的维度变换实现](https://download.csdn.net/download/weixin_38698174/13988496)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v93^chatsearchT3_2"}}] [.reference_item style="max-width: 50%"] - *2* *3* [Pytorch 基础之维度变化](https://blog.csdn.net/zxhandroid/article/details/129192950)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v93^chatsearchT3_2"}}] [.reference_item style="max-width: 50%"] [ .reference_list ]

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值