美文网首页
pytorch:tensor.repeat()

pytorch:tensor.repeat()

作者: 不太聪明的亚子 | 来源:发表于2020-08-07 09:23 被阅读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。

相关文章

网友评论

      本文标题:pytorch:tensor.repeat()

      本文链接:https://www.haomeiwen.com/subject/wohtdktx.html