上传者: 42113794
|
上传时间: 2021-11-26 11:10:26
|
文件大小: 3KB
|
文件类型: -
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 )