在一个循环中追加tensor,不使用append()
时间: 2023-03-31 11:02:20 浏览: 148
在PyTorch中Tensor的查找和筛选例子
5星 · 资源好评率100%
可以使用torch.cat()函数来将多个tensor拼接成一个大的tensor,从而实现在循环中追加tensor的效果。具体实现方法如下:
1. 定义一个空的tensor,用于存储拼接后的结果。
2. 在循环中,每次生成一个新的tensor,将其与之前的结果tensor拼接起来。
3. 将拼接后的结果tensor作为下一次循环的初始值,继续迭代。
示例代码如下:
result = torch.empty() # 定义一个空的tensor
for i in range(10):
new_tensor = torch.randn(3, 4) # 生成一个新的tensor
result = torch.cat((result, new_tensor), dim=) # 将新的tensor与之前的结果tensor拼接起来
print(result) # 输出拼接后的结果tensor
阅读全文