
Compact Bilinear Pooling-Pytorch:基于Pytorch的紧凑型双线性池化实现
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
简介:本项目提供了一个基于Pytorch的实现方案,用于高效计算紧凑型双线性池化,适用于图像和自然语言处理任务,助力模型性能提升。
CompactBilinearPooling-Pytorch 是一个用于实现紧凑型双线性池的 PyTorch 库。通过以下命令安装相关依赖:`pip install pytorch_fft`
使用方法如下:
```python
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)
```
全部评论 (0)
还没有任何评论哟~


