PyTorch获取Tensor大小,size与shape的区别

Posted 算法与编程之美

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了PyTorch获取Tensor大小,size与shape的区别相关的知识,希望对你有一定的参考价值。

问题

Tensor是PyTorch的重要概念之一,获取Tensor的大小的方法主要有两种,其一是size,其二是shape,那么两种方法的区别是什么呢?

方法


import torch

a = torch.rand(size=(3, 224, 224))
# shape是Tensor的一个属性
print(a.shape) # torch.Size([3, 224, 224])

# size()是Tensor的一个方法
print(a.size()) # torch.Size([3, 224, 224])

# 给定参数获取不同的维度size
print(a.size(0)) # 3
print(a.size(1)) # 224 
print(a.size(2)) # 224

结语

以上是关于PyTorch获取Tensor大小,size与shape的区别的主要内容,如果未能解决你的问题,请参考以下文章

pytorch入门与实践-2.2

Pytorch | 详解Pytorch科学计算包——Tensor

pytorch的Tensor的操作

PyTorch-Tensor

Pytorch之Tensor

pytorch