torch.cat((a,b),dim=-1)

_LEON_

关注

阅读 83

2022-03-11

torch.cat((a,b),dim=-1)表示按倒数第一维cat

相关推荐

精彩评论(0)

0 0 举报