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的区别的主要内容,如果未能解决你的问题,请参考以下文章