0
点赞
收藏
分享

微信扫一扫

Pytorch中的repeat()函数

phpworkerman 2022-04-17 阅读 79

参数的size要求和输入向量一致,若TensorA有三维则

TensorA.repeat(?,?,?)代表各自维度重复多少次

A = torch.tensor([[[0, 0],
                   [1, 1],
                   [2, 2]]])

现在重复列2次,两列变四列

 

现在重复行2次,三行变六列

 

同样,现在一个3x2的变成两个3x2的

 

 

举报

相关推荐

0 条评论