ICode9

精准搜索请尝试: 精确搜索
首页 > 其他分享> 文章详细

pytorch 中repeat函数

2021-05-13 13:31:11  阅读:22  来源: 互联网

标签:repeat 函数 shape 张量 维度 pytorch print expand


沿着指定的维度重复tensor。不同与expand(),本函数复制的是tensor中的数据。
扩展(expand)张量不会分配新的内存,只是在存在的张量上创建一个新的视图(view),一个大小(size)等于1的维度扩展到更大的尺寸。repeat沿着特定的维度重复这个张量,和expand()不同的是,这个函数拷贝张量的数据。

import torch
a=torch.linspace(0,12,13)
print(a)
print(a.shape)

在这里插入图片描述

b=a.repeat(13,1)
print(b)
print(b.shape)

在这里插入图片描述

c=b.repeat(3,1,1)
print(c)
print(c.shape)

在这里插入图片描述

标签:repeat,函数,shape,张量,维度,pytorch,print,expand
来源: https://blog.csdn.net/weixin_39289876/article/details/116746557

专注分享技术,共同学习,共同进步。侵权联系[admin#icode9.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有