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

时间:2024-05-22 04:56:16
【文件属性】:

文件名称:CompactBilinearPooling-Pytorch:紧凑型双线性池的Pytorch实现

文件大小:3KB

文件格式:ZIP

更新时间:2024-05-22 04:56:16

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)

网友评论