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。
这个函数如函数名一样,是复制函数,参数表示把这个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。