Tensor张量的属性dim、type、size、shape、numel的使用方法介绍

本文重点

Tensor张量有一些常用的属性，我们可以通过这些基本的属性判断张量的类型，维度，以及元素个数，本节课程我们将对tensor属性进行简单的介绍。

代码

import torch
import numpy as np
a=torch.Tensor(2,3,4)
print(a)
print(a.dim())
print(a.type())
print(a.size())
print(a.shape)
print(a.numel())

常用的属性和方法有：

dim：输出张量的维度

type：张量的类型

size、shape：张量的形状

numel：张量中元素的个数

注意：在后面的很多pytorch的方法中会有很多关于tensor的操作需要指定操作的维度，此时会有一个参数dim，注意dim是从0开始的。

这个程度是干什么的，我们暂时不需要知道，后面会介绍，现在我们只需要知道dim从0开始就ok了