tensor将n个1*768维度的tensor 合并为n*768
时间: 2023-12-21 19:03:22 浏览: 144
tensorboard merged方法
你可以使用 `torch.cat()` 函数将 n 个 1x768 维度的张量合并为 n*768 的张量。这个函数可以在 PyTorch 中使用。以下是一个示例代码:
```python
import torch
# 假设你有 n 个 1x768 的张量
n = 3
tensor_list = [torch.randn(1, 768) for _ in range(n)]
# 使用 torch.cat() 函数进行合并
merged_tensor = torch.cat(tensor_list, dim=0)
# 打印合并后的张量形状
print(merged_tensor.shape)
```
这样,`merged_tensor` 就是一个 n*768 的张量,其中每个元素都是原来的 1x768 张量。请根据你的具体需求替换 `tensor_list`,并根据需要调整维度和张量的形状。
阅读全文