查看一个张量的维度和一个列表的维度

最新推荐文章于 2023-10-27 16:37:24 发布

QT-Smile

最新推荐文章于 2023-10-27 16:37:24 发布

阅读量537

点赞数 1

分类专栏：深度学习之美文章标签： pytorch 深度学习 python

本文链接：https://blog.csdn.net/qq_34848334/article/details/125251069

版权

PyTorch 张量维度列表转换

关键词由CSDN通过智能技术生成

深度学习之美专栏收录该内容

209 篇文章 25 订阅

订阅专栏

查看一个张量的维度和一个列表的维度

# 查看一个张量的维度和一个列表的维度
import torch

a = torch.tensor([[8, 9, 6, 5], [1, 2, 5, 8]])
print("a:", a)
print("a size:", a.size())

b = [[8, 9, 6, 5], [1, 2, 5, 8]]
c = torch.tensor(b)
print("c:", c)
print("c size:", c.size())

print("b:", b)
print("b size:", b.size())

在这里插入图片描述