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(
关于PyTorch中向量的维度
最新推荐文章于 2024-12-14 21:46:21 发布
本文探讨了在PyTorch中遇到的向量维度问题,指出普通tensor向量的shape与转置后的shape相同,导致混淆行向量和列向量。由于维度不匹配,直接将x或x.T输入网络会导致错误。解决方法是正确转换向量的维度,以确保与网络要求的输入匹配,强调在使用中需要注意向量的维度细节。

最低0.47元/天 解锁文章
1107

被折叠的 条评论
为什么被折叠?



