1.torch.cat

官方说明：torch.cat

torch.cat(tensors, dim=0, *, out=None) → Tensor

连接给定维数的给定序列的序列张量。所有张量要么具有相同的形状(除了连接维度)，要么为空。

需要安装 pytorch

2. 代码

import torch
# 总结：
# 1. torch.cat((x,y),dim=0) :张量 X,Y按照列堆起来
# 2. torch.cat((x,y),dim=1) :张量 X,Y按照行并排起来
x = torch.ones(3, 4)
y = torch.zeros(3, 4)
z = torch.cat((x, y), dim=0)  # dim = 0 ;按列堆起来
m = torch.cat((x, y), dim=1)  # dim = 1 :按行并排
print(f'x={x}')
print(f'y={y}')
print(f'z={z}')
print(f'm={m}')

3. 结果

x=tensor([[1., 1., 1., 1.],[1., 1., 1., 1.],[1., 1., 1., 1.]])
y=tensor([[0., 0., 0., 0.],[0., 0., 0., 0.],[0., 0., 0., 0.]])
z=tensor([[1., 1., 1., 1.],[1., 1., 1., 1.],[1., 1., 1., 1.],[0., 0., 0., 0.],[0., 0., 0., 0.],[0., 0., 0., 0.]])
m=tensor([[1., 1., 1., 1., 0., 0., 0., 0.],[1., 1., 1., 1., 0., 0., 0., 0.],[1., 1., 1., 1., 0., 0., 0., 0.]])

4. 总结：

torch.cat((x,y),dim=0) :张量 X,Y按照列堆起来
torch.cat((x,y),dim=1) :张量 X,Y按照行并排起来

torch.cat 关于 dim=0,dim=1 测试相关推荐

torch.sum(),dim=0,dim=1解析
直接上代码: # 定义一个2维张量a a = torch.tensor([[1,2,3],[4,5,6]]) print(a) a.shape output: tensor([[ 1, 2, 3],[ ...
【PyTorch】torch.mean(), dim=0, dim=1
对于torch.mean(s,dim=0),表示跨行求平均. 同理,对于dim=1为第一个维度,代表列. 对于torch.sum(s,dim=1),表示跨列求平均.
PyTorch的torch.cat
字面理解:torch.cat是将两个张量(tensor)拼接在一起,cat是concatnate的意思,即拼接,联系在一起. 例子理解 import torch A=torch.ones(2,3) # ...
Pytorch的torch.cat实例
import torch 通过 help((torch.cat)) 可以查看 cat 的用法 cat(seq,dim,out=None)其中 seq表示要连接的两个序列,以元组的形式给出,例如:seq ...
2021.08.22学习内容torch.cat()和torch.stack()函数
torch.cat()函数将两个张量(tensor)拼接在一起,cat是concatnate的意思,即拼接,联系在一起. def cat(tensors: List[torch.Tensor], d ...
【PyTorch】torch.cat==＞张量拼接，在图像的应用上可以有效利用原始图像结构信息
1. 字面理解:torch.cat是将两个张量(tensor)拼接在一起,cat 是 concatenate 的意思,即拼接,联系在一起. 使用torch.cat((A,B),dim)时,除拼接维数d ...
【pytorch】torch.cat()函数
欢迎移步我的个人博客例子 import torchA=torch.ones(2,3) #2x3的张量(矩阵) Atensor([[ 1., 1., 1.],[ 1., 1., 1.]])B=2*to ...
torch.cat() 函数用法
torch.cat是将两个张量(tensor)拼接在一起,cat是concatnate的意思,即拼接,联系在一起. 使用torch.cat((A,B),dim)时,除拼接维数dim数值可不同外其余维数 ...
pytorch 中 torch.cat 函数的使用
1. 字面理解:torch.cat是将两个张量(tensor)拼接在一起,cat是concatnate的意思,即拼接,联系在一起. 2. 例子理解 >>> import torch ...

torch.cat 关于 dim=0,dim=1 测试

文章目录

1.torch.cat

2. 代码

3. 结果

4. 总结：

torch.cat 关于 dim=0,dim=1 测试相关推荐

最新文章

热门文章