torch.tile用法

Mariooooooooooo

已于 2023-07-10 19:00:39 修改

阅读量1.8k

点赞数 1

分类专栏： Pytorch 文章标签： python 深度学习 pytorch

于 2023-07-10 18:59:06 首次发布

本文链接：https://blog.csdn.net/qq_42256930/article/details/131645897

版权

Pytorch 专栏收录该内容

13 篇文章

订阅专栏

该文展示了如何在PyTorch中利用torch.randint生成随机整数张量，然后用torch.tile函数将张量在行和列上进行复制扩展，从而创建了一个新的张量b。原始张量a是一个5x1的矩阵，经过复制后，b变成了10x3的矩阵。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

import torch
a = torch.randint(0,6,size=(5,1))
print("before copy:",a)
b = torch.tile(a,(2,3))#其中a是待被复制的数据，2是对数据a的行进行扩展,3是对数据a的列维度进行扩展
print("after copy",b)

数据输出为：

before copy: tensor([[1],
        [4],
        [3],
        [4],
        [0]])
  # 数据b的维度为＂［5*2,1*3]=[10,3]
after copy tensor([[1, 1, 1],
        [4, 4, 4],
        [3, 3, 3],
        [4, 4, 4],
        [0, 0, 0],
        [1, 1, 1],
        [4, 4, 4],
        [3, 3, 3],
        [4, 4, 4],
        [0, 0, 0]])