【文件属性】:
文件名称: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)