PyTorch
CozyCode
这个作者很懒,什么都没留下…
展开
-
Pytorch中的unsqueeze与squeeze方法,张量增维与缩维
Pytorch中的unsqueeze函数是用来在张量中添加一个或多个新的维度的。可以通过提供一个包含想要添加的维度位置的列表来一次性添加多个维度。例如,unsqueeze([0, 2])会在0和2的位置添加新的维度。返回的张量和原始张量共享相同的数据。相反,如果想删除多余的维度,可以使用torch.squeeze函数。这个函数会返回一个删除了所有大小为1的维度的张量。假设有一个形状为(3, 4)的张量x,在第1维和第3维添加新的维度,使得张量的形状变为(3, 1, 4, 1)。原创 2023-03-20 19:58:21 · 470 阅读 · 1 评论 -
PyTorch中tensor的几种形状
Scalar 通常就是一个数值 vector 例如: [-5., 2., 0.],在深度学习中通常指特征,例如词向量特征,某一维度特征等 matrix 一般计算的都是矩阵,通常都是多维的 n-dimensional tensor 高维空间 例: import torch from torch import tensor #Scalar x = tensor(11.) x #Out: tensor(11.) x.dim() #Out:0 2 * x #Out:tensor(22.) .原创 2022-01-10 23:37:58 · 453 阅读 · 0 评论