展示如何将一个为形状为(21,)的张量转换为一个21维的行向量:
时间: 2024-03-21 21:40:51 浏览: 141
在PyTorch中,可以使用`tensor.view(1, -1)`函数将一个形状为(21,)的张量转换为一个21维的行向量。具体实现如下:
```python
import torch
# 创建一个形状为(21,)的张量
tensor = torch.randn(21)
# 将张量转换为行向量
row_vector = tensor.view(1, -1)
# 查看行向量的形状
print(row_vector.shape) # 输出:torch.Size([1, 21])
```
在上面的示例代码中,我们首先创建了一个形状为(21,)的张量(使用`torch.randn(21)`生成一个21个随机数的张量)。然后,我们使用`view`函数将该张量转换为一个形状为(1,21)的张量,即一个1行21列的矩阵(行向量)。最后,我们打印输出了该行向量的形状,输出结果为`torch.Size([1, 21])`,即形状为(1,21)的张量,表示一个1行21列的行向量。
阅读全文