1 torch.cat函数
形式
torch.cat(tensors, dim=0, *, out=None)
功能
在指定的维度连接给定序列的张量,所有张量必须具有相同的形状(连接维度除外)或者为空。
参数
- tensors:相同形状的张量序列,非空张量必须具有相同形状(连接维度除外)
- dim:张量连接的维度
使用示例
import torch
if __name__ == '__main__':
a = torch.randn(size=(2, 2))
print(a)
b = torch.randn(size=(2, 2))
print(b)
c = torch.cat((a, b), dim=0)
print(c)
d = torch.cat((a, b), dim=1)
print(d)
输出
tensor([[ 0.8793, 0.3727],
[-2.3334, -1.4567]])
tensor([[-1.0906, -1.2683],
[ 0.7161, -0.5843]])
tensor([[ 0.8793, 0.3727],
[-2.3334, -1.4567],
[-1.0906, -1.2683],
[ 0.7161, -0.5843]])
tensor([[ 0.8793, 0.3727, -1.0906, -1.2683],
[-2.3334, -1.4567, 0.7161, -0.5843]])
本文作者:StubbornHuang
版权声明:本文为站长原创文章,如果转载请注明原文链接!
原文标题:Pytorch – torch.cat函数
原文链接:https://www.stubbornhuang.com/2441/
发布于:2022年12月08日 15:51:48
修改于:2023年06月21日 17:44:48
当前分类随机文章推荐
- Pytorch - masked_fill方法参数详解与使用 阅读1336次,点赞0次
- Pytorch - torch.unsqueeze和torch.squeeze函数 阅读615次,点赞0次
- Pytorch - pad_sequence、pack_padded_sequence、pack_sequence、pad_packed_sequence参数详解与使用 阅读1647次,点赞0次
- Pytorch - 使用pytorch自带的Resnet作为网络的backbone 阅读648次,点赞0次
- Pytorch - torch.nn.Module的parameters()和named_parameters() 阅读811次,点赞0次
- Pytorch - torch.chunk参数详解与使用 阅读1588次,点赞0次
- Pytorch - 多GPU训练方式nn.DataParallel与nn.parallel.DistributedDataParallel的区别 阅读1531次,点赞0次
- Pytorch - 模型断点续训,optimizer.step()报错:RuntimeError Expected all tensors to be on the same device, but found cuda:0 阅读321次,点赞0次
- Pytorch - 梯度累积/梯度累加trick,在显存有限的情况下使用更大batch_size训练模型 阅读1087次,点赞0次
- Pytorch - 检测CUDA、cuDNN以及GPU版本的Pytorch是否安装成功、GPU显存测试 阅读4851次,点赞1次
全站随机文章推荐
- 资源分享 - Digital Lighting & Rendering , Third Edition 英文高清PDF下载 阅读1859次,点赞0次
- Github - 使用新的Personal Access Token进行仓库认证 阅读1977次,点赞0次
- human3.6m : Download(数据集下载) 阅读28809次,点赞40次
- C++ - 线程安全的std::cout 阅读2294次,点赞0次
- 资源分享 - Geometric and Discrete Path Planning for Interactive Virtual Worlds 英文高清PDF下载 阅读1133次,点赞0次
- C++ - vector存储动态指针时正确释放内存 阅读6597次,点赞0次
- 深度学习 - 数据集中训练集、验证集、测试集的划分以及各自作用 阅读3309次,点赞1次
- 资源分享 - Artificial Intelligence for Games , First Edition 英文高清PDF下载 阅读1339次,点赞0次
- Pytorch - torch.unsqueeze和torch.squeeze函数 阅读615次,点赞0次
- WordPress - wp_mail发送邮件失败,使用插件或者纯代码方式添加SMTP邮件发送功能 阅读1448次,点赞0次
评论
169