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的物理存储必须是

最低0.47元/天 解锁文章
4694

被折叠的 条评论
为什么被折叠?



