[人工智能] PyTorch - torch.Tensor.repeat

开发: C++知识库 Java知识库 JavaScript Python PHP知识库人工智能区块链大数据移动开发嵌入式开发工具数据结构与算法开发测试游戏开发网络协议系统运维
教程: HTML教程 CSS教程 JavaScript教程 Go语言教程 JQuery教程 VUE教程 VUE3教程 Bootstrap教程 SQL数据库教程 C语言教程 C++教程 Java教程 Python教程 Python3教程 C#教程
数码: 电脑笔记本显卡显示器固态硬盘硬盘耳机手机 iphone vivo oppo 小米华为单反装机图拉丁

-> 人工智能 -> PyTorch - torch.Tensor.repeat -> 正文阅读

[人工智能]PyTorch - torch.Tensor.repeat

作者:recommend-item-box type_blog clearfix

PyTorch - torch.Tensor.repeat

flyfish

torch.Tensor.repeat
沿指定维度重复此张量

import torch
# 可以用行列来描述，更容易理解，如何重复[1, 2, 3]的问题
a = torch.tensor([1, 2, 3])
#tensor([1, 2, 3])
#如何重复[1, 2, 3]的问题
print(a.repeat(2)) # 2 * 原列数
#tensor([1, 2, 3, 1, 2, 3])

print(a.repeat(1,2)) #里面的数与a.repeat(2)相比没变，但是维度增加了。1* 原行数，2 * 原列数
#tensor([[1, 2, 3, 1, 2, 3]])

print(a.repeat(3, 2)) #3* 原行数，2 * 原列数
# tensor([[1, 2, 3, 1, 2, 3],
#         [1, 2, 3, 1, 2, 3],
#         [1, 2, 3, 1, 2, 3]])
print(a.repeat(4, 3, 2)) # 可以理解成4个  （3 * 原行数，2 * 原列数）
# tensor([[[1, 2, 3, 1, 2, 3],
#          [1, 2, 3, 1, 2, 3],
#          [1, 2, 3, 1, 2, 3]],
#
#         [[1, 2, 3, 1, 2, 3],
#          [1, 2, 3, 1, 2, 3],
#          [1, 2, 3, 1, 2, 3]],
#
#         [[1, 2, 3, 1, 2, 3],
#          [1, 2, 3, 1, 2, 3],
#          [1, 2, 3, 1, 2, 3]],
#
#         [[1, 2, 3, 1, 2, 3],
#          [1, 2, 3, 1, 2, 3],
#          [1, 2, 3, 1, 2, 3]]])

print(a.repeat(4, 3, 2).shape)
# torch.Size([4, 3, 6])