torch.cat的dim-1
时间: 2024-03-29 08:33:41 浏览: 49
torch-1.0.0
5星 · 资源好评率100%
torch.cat函数是PyTorch中用于拼接张量的函数。它可以将多个张量按照指定的维度进行拼接。当dim参数设置为-1时,表示在最后一个维度上进行拼接。
具体来说,如果有两个形状相同的张量A和B,它们的形状为[batch_size, feature_dim],那么使用torch.cat([A, B], dim=-1)将会得到一个形状为[batch_size, 2*feature_dim]的新张量。
阅读全文