关于PyTorch中向量的维度

龙哥每天都要学

于 2021-07-26 00:14:49 发布

阅读量947

点赞数

CC 4.0 BY-SA版权

分类专栏： Pytorch深度学习文章标签：人工智能深度学习机器学习 pytorch python

本文链接：https://blog.youkuaiyun.com/rambo_csdn_123/article/details/119090463

本文探讨了在PyTorch中遇到的向量维度问题，指出普通tensor向量的shape与转置后的shape相同，导致混淆行向量和列向量。由于维度不匹配，直接将x或x.T输入网络会导致错误。解决方法是正确转换向量的维度，以确保与网络要求的输入匹配，强调在使用中需要注意向量的维度细节。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

import torch
x = torch.tensor([1, 2, 3, 4], dtype=torch.float)
x.shape
Out[14]: torch.Size([4])
y = x.T
y.shape
Out[16]: torch.Size([<