torch x = x.view(x.size(0),-1)的理解

目录

一、概述

二、代码展示


一、概述

1.在torch的网络定义部分的forward(self,x)中有可能会出现下面这句话:

 x = x.view(x.size(0),-1)

2.这句话的意思是将多维度的Tensor展平成一维,但是到底转换的样子是什么样的其实我们这样看并不明白

二、代码展示

1.我们先定义一个Tensor:

import torch

a = torch.arange(1,17)

2.输出这个Tensor的内容和大小:

print(a)
print(a.size(0))

代码变为:

import torch

a = torch.arange(1,17)

print(a)
print(a.size(0))

结果为:

                               

3.使用

 x = x.view(x.size(0),-1)

将Tensor转换为1维

输入代码:

b = a.view(a.size(0),-1)
print(b)

代码变为:

import torch

a = torch.arange(1,17)

print(a)
print(a.size(0))

b = a.view(a.size(0),-1)
print(b)

结果为;

                                    

其中这个-1指的是不知道多少列的情况下,根据原来Tensor内容和Tensor的大小自动分配列数

4.换句话,和我们这样做的效果是一样的,即指定列数为1:

c = a.view(a.size(0),1)
print(c)

代码变为:

import torch

a = torch.arange(1,17)

print(a)
print(a.size(0))

b = a.view(a.size(0),-1)
print(b)
c = a.view(a.size(0),1)
print(c)

结果为:

                                     

结果果然是一样的。

5.我们还可以将行数不定(为1)指定列数为Tensor的大小

d = a.view(-1, a.size(0))
print(d)

代码变为:

import torch

a = torch.arange(1,17)

print(a)
print(a.size(0))

b = a.view(a.size(0),-1)
print(b)
c = a.view(a.size(0),1)
print(c)

d = a.view(-1, a.size(0))
print(d)

结果为;

                                   

评论 8
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

老兵安帕赫

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值