pytorch tensor flip

最新推荐文章于 2024-05-30 10:06:22 发布

落花逐流水

最新推荐文章于 2024-05-30 10:06:22 发布

阅读量1.5k

点赞数

分类专栏： pytorch实践文章标签： pytorch python 深度学习

本文链接：https://blog.csdn.net/juluwangriyue/article/details/119987125

版权

136 篇文章 114 订阅 ¥29.90 ¥99.00

订阅专栏

本文介绍了如何在PyTorch中使用torch.flip函数对数据进行翻转操作，对比了Python numpy的flip方法和C++中的实现，并提供了相关示例及官方文档链接，适合深度学习开发者参考。

摘要由CSDN通过智能技术生成

因为使用中用到数据需要flip，在python代码中使用numpy的flip很方便，但是c++中使用flip就不那么方便了。
本来图像数据，想要使用vtk去做数据处理，后来发现使用pytoch的torch处理更合适一些。

torch.flip(input,dim):第一个参数是输入，第二个参数是输入的第几维度，按照维度对输入进行翻转

# -*- coding : UTF-8 -*-
# @file   : torch_t1.py
# @Time   : 2021/8/29 22:37
# @Author : wmz
import torch
x = torch.arange(16).view(2, 2

了解本专栏

关注