So if A and B are of shape (3, 4), torch.cat([A, B], dim=0) will be of ... here is the link from the pytorch forum with discussions on this: ...
確定! 回上一頁