PyTorch提供了大量的对Tensor进行操作的函数或方法
这些函数内部使用指针实现对矩阵的形状变化、拼接和拆分等操作
1、Tensor.nelement
Tensor.nelement用来查看矩阵元素的个数
例:
输出为:
2、Tensor.ndimension
Tensor.ndimension用来查看矩阵元素的轴的个数
例:
输出为:
3、ndimension.size
ndimension.size用来查看矩阵元素的维度
属性Tensor.shape也可以用来查看Tensor的维度
例:
输出为:
4、Tensor.view和Tensor.reshape
PyTorch中,Tensor.view和Tensor.reshape都能被用来更改Tensor的维度。
区别:
1、Tensor.view
Tensor.view要求Tensor的物理存储必须是