pytorch: 获取 tensor 维度的方法:shape 和 size()

pycharm 获取 tensor 的方法有两种:shape 和 size()
tensor 是类 Tensor() 的实例, 其中shape是其属性,而 size() 是其继承的方法,两者均可以获得 tensor 的维度。

import torch

a = torch.tensor([[1, 2, 3], [4, 5, 6]])
print(a.shape)
print(a.size())

输出为:
torch.Size([2, 3])
torch.Size([2, 3])

另外,还可以获取其中的某一维度

print(a.shape[0])
print(a.shape[1])
print(a.size(0))
print(a.size(1))

输出为:

2
3
2
3

重点:
只要注意其用法就行了,shape 是属性,使用中括号,size() 是函数,使用小括号。

  • 3
    点赞
  • 5
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
### 回答1: PyTorch中有多种方法可以用来压缩和减小Tensor维度,以下是其中一些常用的方法: 1. squeeze()方法:squeeze()方法可以将Tensor维度为1的维度去除。例如,如果有一个维度为[1,3,1,5]的Tensor,使用squeeze()方法后,它的维度将变为[3,5]。使用squeeze()方法的代码示例如下: ``` import torch x = torch.randn(1, 3, 1, 5) y = x.squeeze() print(y.size()) ``` 2. unsqueeze()方法:unsqueeze()方法可以在Tensor中插入新的维度。例如,如果有一个维度为[3,5]的Tensor,使用unsqueeze()方法后,它的维度将变为[1,3,1,5]。使用unsqueeze()方法的代码示例如下: ``` import torch x = torch.randn(3, 5) y = x.unsqueeze(0) print(y.size()) ``` 3. view()方法:view()方法可以用于改变Tensor维度,但是要保证Tensor中元素的总数不变。例如,如果有一个维度为[3,5]的Tensor,使用view(1, 1, 3, 5)方法后,它的维度将变为[1,1,3,5]。使用view()方法的代码示例如下: ``` import torch x = torch.randn(3, 5) y = x.view(1, 1, 3, 5) print(y.size()) ``` 4. reshape()方法:reshape()方法也可以用于改变Tensor维度,但是与view()方法不同的是,reshape()方法可以改变Tensor中元素的总数。例如,如果有一个维度为[3,5]的Tensor,使用reshape(1, 1, 15)方法后,它的维度将变为[1,1,15]。使用reshape()方法的代码示例如下: ``` import torch x = torch.randn(3, 5) y = x.reshape(1, 1, 15) print(y.size()) ``` 这些方法可以根据不同的需求,灵活地压缩和减小Tensor维度。 ### 回答2: 在PyTorch中,可以使用squeeze()函数来压缩Tensor维度。squeeze()函数可以去除Tensor维度为1的维度,从而达到压缩Tensor维度的效果。 具体用法如下: ``` import torch # 创建一个Tensor维度为(1, 3, 1, 5) x = torch.randn(1, 3, 1, 5) # 使用squeeze()函数压缩维度 # 压缩后的维度为(3, 5) x_squeezed = x.squeeze() print(x.shape) # torch.Size([1, 3, 1, 5]) print(x_squeezed.shape) # torch.Size([3, 5]) ``` 在上述代码中,首先创建了一个维度为(1, 3, 1, 5)的Tensor。然后使用squeeze()函数压缩了Tensor维度。最后打印了压缩前后的Tensor维度。 需要注意的是,squeeze()函数默认会压缩所有维度为1的维度,如果希望只压缩指定的维度,可以使用squeeze(dim)函数。其中dim表示要压缩的维度的索引。 例如,如果只想压缩第二个维度(索引为1)的维度为1的维度,可以像下面这样操作: ``` import torch # 创建一个Tensor维度为(1, 3, 1, 5) x = torch.randn(1, 3, 1, 5) # 使用squeeze(dim)函数压缩指定维度 # 压缩后的维度为(1, 3, 5) x_squeezed = x.squeeze(2) print(x.shape) # torch.Size([1, 3, 1, 5]) print(x_squeezed.shape) # torch.Size([1, 3, 5]) ``` 在上述代码中,squeeze(2)表示只压缩第二个维度(索引为2)的维度为1的维度。输出的Tensor维度为(1, 3, 5)。 ### 回答3: 在PyTorch中,可以使用squeeze()和unsqueeze()这两个函数来压缩和扩展Tensor维度。 squeeze()函数用于压缩Tensor维度为1的维度。例如,假设有一个形状为(1, 3, 1, 4)的Tensor,在第0和第2维度上的维度为1,可以使用squeeze()函数将其压缩为(3,4)的形状。具体操作如下: ```python import torch x = torch.randn(1, 3, 1, 4) print(x.shape) # 输出:torch.Size([1, 3, 1, 4]) y = x.squeeze() print(y.shape) # 输出:torch.Size([3, 4]) ``` unsqueeze()函数用于在Tensor中插入维度为1的维度。例如,假设有一个形状为(3, 4)的Tensor,可以使用unsqueeze()函数在指定位置插入维度为1的维度。具体操作如下: ```python import torch x = torch.randn(3, 4) print(x.shape) # 输出:torch.Size([3, 4]) y = x.unsqueeze(0) print(y.shape) # 输出:torch.Size([1, 3, 4]) z = x.unsqueeze(1) print(z.shape) # 输出:torch.Size([3, 1, 4]) w = x.unsqueeze(2) print(w.shape) # 输出:torch.Size([3, 4, 1]) ``` 使用squeeze()和unsqueeze()函数可以方便地对Tensor进行压缩和扩展操作,便于进行后续的计算或处理。

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值