python如何创建四维tensor
时间: 2023-05-28 21:05:24 浏览: 230
python生成四叉树
在PyTorch中,可以使用torch.Tensor()函数创建四维张量。下面是一个示例代码:
```python
import torch
# 创建一个4维张量,形状为(2, 3, 4, 5)
x = torch.Tensor(2, 3, 4, 5)
# 打印张量形状
print(x.shape)
```
输出:
```
torch.Size([2, 3, 4, 5])
```
在这个例子中,我们创建了一个形状为(2, 3, 4, 5)的四维张量x。第一个维度有2个元素,第二个维度有3个元素,第三个维度有4个元素,第四个维度有5个元素。张量中的每个元素都是随机初始化的。
阅读全文