随机生成a,b,c三个tensor数据,将a,b,c的tensor尺寸大小以字典输出
时间: 2023-05-11 17:03:46 浏览: 90
可以使用以下代码生成a,b,c三个随机的tensor数据,并将它们的尺寸大小以字典输出:
```python
import torch
a = torch.randn(3, 4)
b = torch.randn(2, 5)
c = torch.randn(6, 1)
sizes_dict = {'a': tuple(a.size()), 'b': tuple(b.size()), 'c': tuple(c.size())}
print(sizes_dict)
```
输出结果为:
```
{'a': (3, 4), 'b': (2, 5), 'c': (6, 1)}
```
注意,这里使用了PyTorch库来生成随机的tensor数据。
阅读全文