torch.cat和torch.concat有什么区别
时间: 2023-04-05 22:03:57 浏览: 128
PyTorch的torch.cat用法
5星 · 资源好评率100%
torch.cat和torch.concat都是PyTorch中用于拼接张量的函数,但是它们的参数和用法略有不同。torch.cat接受一个张量序列作为输入,可以在任意维度上拼接张量,而torch.concat则需要指定拼接的维度。另外,torch.concat还可以指定是否在拼接维度上进行拷贝操作。
阅读全文