本文重点
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了