如何将PyTorch中的tensor尺寸修改为长尾词形式?
- 内容介绍
- 文章标签
- 相关推荐
本文共计488个文字,预计阅读时间需要2分钟。
改变Tensor形状的操作:使用tensor.view方法,可以调整tensor的形状,但需保证调整前后元素总数一致。view不会改变原始数据,返回的新tensor与原tensor共享内存,修改其中一个会影响另一个。
改变Tensor尺寸的操作
1.tensor.view
tensor.view方法,可以调整tensor的形状,但必须保证调整前后元素总数一致。view不会改变自身数据,返回的新的tensor与源tensor共享内存,即更改其中一个,另外一个也会跟着改变。
例:
In: import torch as t a = t.arange(0, 6) a.view(2, 3) Out:tensor([[0, 1, 2], [3, 4, 5]]) In: b = a.view(-1, 3)#当某一维为-1的时候,会自动计算它的大小 Out:tensor([[0, 1, 2], [3, 4, 5]])
2.tensor.unsqueeze 和 tensor.squeeze
tensor.unsqueeze 和 tensor.squeeze分别用于增加或减少tensor的某一维度。
本文共计488个文字,预计阅读时间需要2分钟。
改变Tensor形状的操作:使用tensor.view方法,可以调整tensor的形状,但需保证调整前后元素总数一致。view不会改变原始数据,返回的新tensor与原tensor共享内存,修改其中一个会影响另一个。
改变Tensor尺寸的操作
1.tensor.view
tensor.view方法,可以调整tensor的形状,但必须保证调整前后元素总数一致。view不会改变自身数据,返回的新的tensor与源tensor共享内存,即更改其中一个,另外一个也会跟着改变。
例:
In: import torch as t a = t.arange(0, 6) a.view(2, 3) Out:tensor([[0, 1, 2], [3, 4, 5]]) In: b = a.view(-1, 3)#当某一维为-1的时候,会自动计算它的大小 Out:tensor([[0, 1, 2], [3, 4, 5]])
2.tensor.unsqueeze 和 tensor.squeeze
tensor.unsqueeze 和 tensor.squeeze分别用于增加或减少tensor的某一维度。

