怎么打印tensor的shape
时间: 2023-12-19 17:04:09 浏览: 57
使用tensorflow根据输入更改tensor shape
可以使用Tensor的shape属性来打印tensor的shape,例如:
```python
import torch
x = torch.randn(3, 4, 5)
print(x.shape)
```
输出结果为:
```
torch.Size([3, 4, 5])
```
其中torch.Size是一个元组,包含了tensor的每个维度的大小。
阅读全文