torch.size(0)，torch.size(1)中0和1的意思

最新推荐文章于 2024-07-17 14:59:12 发布

原创

最新推荐文章于 2024-07-17 14:59:12 发布

·

1.9w 阅读

·

27

·

59

·

CC 4.0 BY-SA版权

版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。

文章标签：

#pytorch

#深度学习

本文介绍了PyTorch中张量torch.tensor的size()方法，通过示例解释了size(0)和size(1)分别表示张量第0维和第1维的数据数量。例如，对于二维张量a，a.size(0)返回其行数，a.size(1)返回列数。

部署运行你感兴趣的模型镜像一键部署

torch.size(0)中的0表示第0维度的数据数量，torch.size(1)同理。

code：

a = torch.tensor([[1,2,3], [4,5,6]])

print(a.size(0)) # 2 第0维有[1,2,3]和[4,5,6]两个数据

print(a.size(1)) # 3 第1维有1，2，3（或4，5，6）三个数据

您可能感兴趣的与本文相关的镜像

PyTorch 2.5

PyTorch

Cuda

PyTorch 是一个开源的 Python 机器学习库，基于 Torch 库，底层由 C++ 实现，应用于人工智能领域，如计算机视觉和自然语言处理

一键部署运行