torch输出张量的大小
时间: 2024-03-09 12:46:30 浏览: 71
pytorch 计算ConvTranspose1d输出特征大小方式
可以通过张量的属性`size()`或者`shape`来输出张量的大小,例如:
```python
import torch
# 创建一个大小为(2, 3, 4)的三维张量
x = torch.randn(2, 3, 4)
# 输出张量的大小
print(x.size())
print(x.shape)
```
输出结果为:
```
torch.Size([2, 3, 4])
torch.Size([2, 3, 4])
```
阅读全文