Tensor有多种创建方法,如基础的构造函数Tensor(),还有多种与NumPy十分类似的方法,如ones()、eye()、zeros()和randn()等,下图列举了常见的Tensor创建方法:
从代码角度实现Tensor的多种创建方法:
对于Tensor的维度,可使用Tensor.shape或者size()函数查看每一维的大小,两者等价
查看Tensor中的元素个数,可使用Tensor.numel或者Tensor.nelement()函数,两者等价
Tensor有多种创建方法,如基础的构造函数Tensor(),还有多种与NumPy十分类似的方法,如ones()、eye()、zeros()和randn()等,下图列举了常见的Tensor创建方法:
从代码角度实现Tensor的多种创建方法:
对于Tensor的维度,可使用Tensor.shape或者size()函数查看每一维的大小,两者等价
查看Tensor中的元素个数,可使用Tensor.numel或者Tensor.nelement()函数,两者等价