打印tensor的shape
时间: 2023-08-24 19:09:47 浏览: 46
在PyTorch中,可以使用`shape`属性来打印tensor的形状。示例代码如下:
```python
import torch
# 创建一个2x3的tensor
x = torch.Tensor([[1, 2, 3], [4, 5, 6]])
# 打印tensor的形状
print(x.shape)
```
运行结果为:
```
torch.Size([2, 3])
```
这表示`x`是一个2行3列的矩阵。`torch.Size`是一个元组,其中第一个元素表示行数,第二个元素表示列数,以此类推。
相关推荐
![pdf](https://img-home.csdnimg.cn/images/20210720083512.png)
![pdf](https://img-home.csdnimg.cn/images/20210720083512.png)
![pdf](https://img-home.csdnimg.cn/images/20210720083512.png)
![pdf](https://img-home.csdnimg.cn/images/20210720083512.png)
![pdf](https://img-home.csdnimg.cn/images/20210720083512.png)
![pdf](https://img-home.csdnimg.cn/images/20210720083512.png)
![pdf](https://img-home.csdnimg.cn/images/20210720083512.png)
![pdf](https://img-home.csdnimg.cn/images/20210720083512.png)
![pdf](https://img-home.csdnimg.cn/images/20210720083512.png)
![pdf](https://img-home.csdnimg.cn/images/20210720083512.png)
![xlsx](https://img-home.csdnimg.cn/images/20210720083732.png)
![pdf](https://img-home.csdnimg.cn/images/20210720083512.png)
![zip](https://img-home.csdnimg.cn/images/20210720083736.png)
![zip](https://img-home.csdnimg.cn/images/20210720083736.png)
![zip](https://img-home.csdnimg.cn/images/20210720083736.png)