栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

7.libtorch张量的合并与拆分

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

7.libtorch张量的合并与拆分

表 2-6. 合并与拆分
函数 描述
torch::hstack 沿着水平方向(1维度)拼接
torch::vstack 沿着竖直方向(0维度)拼接
torch::dstack 沿着第2维度方向(2维度)拼接
torch::cat 将两个张量(tensor)拼接在一起,cat是concatenate的意思,即拼接,联系在一起
torch::chunk(intput,chunks,dim=0) 把一个tensor均匀分割成若干个小tensor。
torch::split(tensor,split_size,dim=0) 沿着某维度按split_size进行分割。

例2-12:合并与拆分

#include 
#include 
#include 
using namespace torch::indexing;

int main() {
	auto a = torch::zeros({ 2,3 });
	auto b = torch::ones({4,3});
	auto c = torch::range(1, 4, 1).reshape({ 2,2 });
	auto d = torch::range(1, 50, 1).reshape({ 5,5,2 });
	auto e = torch::r
转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/498910.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号