input_tensor = out[1];target_tensor = out[2];用pytorch把这两个tensor拼接起来。
时间: 2023-03-25 10:04:06 浏览: 139
PyTorch中Tensor的拼接与拆分的实现
5星 · 资源好评率100%
可以使用torch.cat()函数将这两个tensor拼接起来,代码如下:
concat_tensor = torch.cat((input_tensor, target_tensor), dim=)
阅读全文