pytorch:tensor.repeat()

2020-08-07  本文已影响0人  不太聪明的亚子

这个函数如函数名一样,是复制函数,参数表示把这个tensor复制成多少个,参数以1,2,3位来解释:

假设a是一个tensor,那么把a看作最小单元:

a.repeat(2)表示在复制1行2列a;

a.repeat(3, 2)表示复制3行2列个a;

a.repeat(3, 2, 1)表示复制3个2行1列个a。

上一篇 下一篇

猜你喜欢

热点阅读