补充火炬功能的小功能集合
torchure的Python项目详细描述
托奇芬(原托奇尔)
##对于那些希望torch提供更方便功能的用户
功能
- 展平(张量)
- imshow(张量)
- 加载
安装torchfun
pip install torchfun
API
扁平(模块)
用于重塑输出
用法:
flat = Flatten()
out = flat(x)
扁平化(功能)
用法:
out = flatten(x)
imshow(功能)
仅处理火炬通道第一批图像,
参数:
- X:输入数据立方体、火炬张量或核弹阵列。
- 标题:将标题添加到绘图。(默认无)
- 标题可以是字符串,也可以是任何可字符串的对象。
- 自动关闭:(默认为真)
- 之后关闭pyplot会话。
- 清洁环境就像你从未在这里使用matplotlib一样。
- 如果设置为False,则绘图将保留在内存中以供进一步绘制。
用法:
IMSHOW(批量) imshow(批处理,标题=[A,B,C]) imshow(批处理,title='title') imshow(批处理,自动关闭=false)
负荷(功能)
参数:
名为
a
和b
的任意参数 将权重a
加载到模型b
,或使用权重a
加载模型b
。 争论的顺序无关紧要。 示例:load('weights.pts',model)
或
>load(model,'weights.pts')
或
>f = open('weight.pts')
>load(f,model)
或
>load(model,f)
返回值:
- 无