jiafei828的博客

原创 pytorch tensor一些操作

文章目录前言一些概念tensortensor维度调整tensor堆叠tensor筛选tensor排序tensor替换数值tensor分割其他操作 前言 本博客主要用于记录本人学习pytorch过程中一些笔记，并非零基础入门，方便自己查询，主要是深度学习计算机视觉方向。 一些概念 深度学习模型输入一般为数个图像构成的张量，张量shape为**[B, C, H, W]**的形式。 torch中分类损失函数如CrossEntropyLoss()、BCELoss()，会自动对输入标签进行进行one-hot。