CompactBilinearPooling-Pytorch:紧凑型双线性池的Pytorch实现

时间:2021-05-02 11:09:36
【文件属性】:
文件名称:CompactBilinearPooling-Pytorch:紧凑型双线性池的Pytorch实现
文件大小:3KB
文件格式:ZIP
更新时间:2021-05-02 11:09:36
Python CompactBilinearPooling-Pytorch 紧凑型双线性池的Pytorch实现。 改编自 先决条件 通过安装 pip install pytorch_fft 用法 from torch import nn from torch . autograd import Variable from CompactBilinearPooling import CompactBilinearPooling bottom1 = Variable ( torch . randn ( 128 , 512 , 14 , 14 )). cuda () bottom2 = Variable ( torch . randn ( 128 , 512 , 14 , 14 )). cuda () layer = CompactBilinearPooling ( 512 , 512 , 8000 )
【文件预览】:
CompactBilinearPooling-Pytorch-master
----CompactBilinearPooling.py(5KB)
----README.md(1002B)

网友评论