pytorch的expand_as和expand

expand(*sizes) → Tensor
参数：sizes的数组，可以是list，tuple，或者直接输入n, m
例子：
expand_as(other) → Tensor
参数：tensor

b = torch.arange(0, 6).view(6, 1, 1)
print(b.shape)  # torch.Size([6, 1, 1])c = torch.Tensor(6, 2, 6)
print(c.shape)d = b.expand_as(c)
print(d.shape) # torch.Size([6, 2, 6])e = b.expand((6, 2, 6)) # expand传入的是数组，比如list, tuple
e = b.expand([6, 2, 6]) # 正确
e = b.expand(6, 2, 6) # 正确
# e = b.expand(torch.Tensor(6, 2, 6)) # expand传入的是数组，不是tensor,这个报错
print(e.shape) # torch.Size([6, 2, 6])

注意

很多人容易出错的地方是：源tensor和传入的矩阵或tensor维度在不为1的维度上必须相同，比如源source_tensor.size=(6, 1, 1)，但是传入的size=(4, 1, 2)，6和4不同报错，但是如果改成
source_tensor.size=(6, 1, 1)，传入的size=(6, 1, 2)或(6, 4, 9)都不会报错，因为在需要expand的维度上，原始tensor的维度值为1，可以expand或expand_as成任何维度。也就是说，不为1的维度值上的维度是不能被expand或expand_as的。

pytorch的expand_as和expand相关推荐

Pytorch中expand_as()函数详解
关于expand_as()的个人理解 from __future__ import print_function import torch as t# a = t.Tensor(2,3) # # pr ...
pytorch之tensor操作expand
import torch# expand 将tensor按照某一个维度扩大或者将tensor整体扩大 # expand 不会额外分配空间,只是原tensor的一个视图x = torch.tensor( ...
关于squeeze unsqueeze 以及expand的学习
参考链接因为自己之前对于squeeze 以及unsqueeze应用较多,这里不再赘述,只给一个简单的例子 >>> import torch >>> a=torch ...
从零开始行人重识别学习笔记
原文:从零开始行人重识别 (本文在原文的基础上增加了一些操作步骤截图.代码注释等等一些自己做的笔记补充,由于本人是新手,第一次接触reid和pytorch,有错误谢谢指出) 准备步骤:Anaconda ...
pytorch中的expand()和expand_as()函数--扩展张量中某维数据的尺寸
pytorch中的expand()和expand_as()函数 1.expand()函数: (1)函数功能: expand()函数的功能是用来扩展张量中某维数据的尺寸,它返回输入张量在某维扩展为更大尺 ...
Pytorch中tensor.expand()和tensor.expand_as()函数
Pytorch中tensor.expand函数 Tensor.expand()函数详解 Tensor.expand_as()函数 Tensor.expand()函数详解函数语法: # 官方解释: D ...
【Pytorch】torch.Tensor.expand_as()与torch.Tensor.expand()使用与比较
torch.Tensor.expand_as官方文档地址:https://pytorch.org/docs/stable/generated/torch.Tensor.expand_as.html?h ...
【PyTorch基础】——expand(）和expand_as()
1.expand()函数功能: 扩展张量中某维数据的尺寸,返回输入张量在某维扩展为更大尺寸后的张量,且原始tensor和扩展后tensor不共享内存. 参数: 括号中输入参数为指定经过维度尺寸扩展后 ...
pytorch中expand()和expand_as()和repeat()函数解读
简要三个函数都是不扩展维度却改变tensor维度数值存在的.关于扩展维度查看squeeze和unsqueeze;关于更改维度位置查看transpose和 permute 1. expand()和ex ...

pytorch的expand_as和expand

注意

pytorch的expand_as和expand相关推荐

最新文章

热门文章