torch输出张量的维度
时间: 2023-08-29 09:08:12 浏览: 87
PyTorch中Tensor的维度变换实现
5星 · 资源好评率100%
可以通过张量的属性`dim()`或者`ndim`来输出张量的维度,例如:
```python
import torch
# 创建一个大小为(2, 3, 4)的三维张量
x = torch.randn(2, 3, 4)
# 输出张量的维度
print(x.dim())
print(x.ndim)
```
输出结果为:
```
3
3
```
阅读全文