使用pytorch生成一个224*224*3的张量
时间: 2023-12-05 21:05:48 浏览: 86
PyTorch中张量的创建方法
可以使用torch.randn()函数生成一个指定大小的随机张量,然后将其乘以255并转换为整型,得到一个224*224*3的张量。具体实现如下:
```python
import torch
# 生成一个224*224*3的张量
tensor = torch.randn(224, 224, 3) * 255
tensor = tensor.int()
print(tensor)
```
阅读全文