补充火炬功能的小功能集合

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)

负荷(功能)

参数:

  • 名为ab的任意参数 将权重a加载到模型b,或使用权重a加载模型b。 争论的顺序无关紧要。 示例:

    load('weights.pts',model)

>load(model,'weights.pts')

>f = open('weight.pts')
>load(f,model)

>load(model,f)

返回值:

欢迎加入QQ群-->: 979659372 Python中文网_新手群

推荐PyPI第三方库


热门话题
java与Groovy正则表达式匹配混淆   java控制台未定义为JSNI   主目录中的java指针异常   java如何获取hashmap中的arrayList的大小   ApachePOI用于文本,Java代码中的word doc(.docx)中没有进行追加   Ruby对象到Java对象   JavaSpringJPA哈希集只返回一个值   Java在执行由配置了精确类路径参数的mavenjarpluin生成的可执行jar时无法找到依赖项   java我试图在画布上写一个文本,但什么也没发生   java HTTP 404源服务器找不到目标资源的当前表示形式,或者不愿意透露存在该表示形式。不起作用   java提取小数点后的最后一个数字   oop Java:无法访问对象的元素   PHP URL中RecyclerView中的java句柄空异常   不调用OnCreateViewHolder的java Update RecyclerView适配器项