深度学习框架-pytorch
文章平均质量分 66
pytorch学习记录
Bingo
不爱串门的老王
展开
-
dyn-unet对训练好的模型进行推理输出结果
import osimport torchimport SimpleITK as sitkimport numpy as npfrom monai.networks.nets import UNet, DynUNetfrom monai.transforms import ( AsDiscrete, AddChannel, Compose, ToTensor, SqueezeDim, Activations, Resize, Spac原创 2021-04-28 10:06:02 · 1274 阅读 · 1 评论 -
pytorch中改变维度操作
最近对网络做改进,遇到了一些改变tensor数据维度的操作,特记录在此,方便以后查阅。1.使用索引改变维度值a = torch.randint(10, (1, 3, 4, 5, 6))b = a[:, 1:]c = a[:, 1:, :, :, :]print(a.shape)print(b.shape)print(c.shape)输出结果:这里可以看到,“:”在python中含有任取的意思。当我们索引的时候。如果再某个维度上使用了:,则说明这个维度上的数据我们可以随便取(我全都要原创 2021-09-01 15:47:58 · 4704 阅读 · 0 评论 -
pytorch中的einsum函数
einsum函数einsum函数在文档中的解释为:即,该函数提供了一种使用爱因斯坦求和约定来计算多线性表达式的方法(即乘积和)爱因斯坦求和约定(Einstein summation convention)关于爱因斯坦求和约定的详细运算过程参考知乎大佬的回答:https://zhuanlan.zhihu.com/p/46006162以及B站up的视频:https://www.bilibili.com/video/BV1ot411V71V?from=search&seid=9196910873原创 2021-07-21 15:00:19 · 2238 阅读 · 0 评论