当前位置:主页 > 软件编程 > Python代码 >

对PyTorch torch.stack的实例讲解

时间:2021-03-27 09:29:49 | 栏目:Python代码 | 点击:

不是concat的意思

import torch
a = torch.ones([1,2])
b = torch.ones([1,2])
torch.stack([a,b],1)
(0 ,.,.) = 
 1 1
 1 1
[torch.FloatTensor of size 1x2x2]

您可能感兴趣的文章:

相关文章