一维数组变为tensor的shape就是一维的！！

最新推荐文章于 2024-08-21 13:01:35 发布

王旺旺！

最新推荐文章于 2024-08-21 13:01:35 发布

阅读量398

点赞数 3

文章标签： pytorch 深度学习 python

本文链接：https://blog.csdn.net/hhhhhh356557/article/details/136513623

版权

本文展示了如何在PyTorch中处理类标签数据，包括创建`torch.tensor`表示class_labels，以及使用`torch.cat`函数合并有标签数据y和无标签数据y_null，最终调整数据形状。

摘要由CSDN通过智能技术生成

class_labels = [207, 360, 387, 974, 88, 979, 417, 279]
y = torch.tensor(class_labels)
print(y.shape)
y_null = torch.tensor([1000] * 8)  # n个1000，其实应该没有这个1000的标签了  0-999  它用来做无标签的形式
y = torch.cat([y, y_null], 0)
print(y.shape)