onnx runtime fastapi模型部署

fastapi 模型部署大体流程如下： * torch-> onnx * fastapi做为接口服务，后端使用onnx推理 * Docker打包部署 fastapi官…

squeeze torch数据降维

``` import torch import torch.optim as optim from torch.nn import functional as F from torch.utils.data import DataLoader, Dataset ``` ``` y = torch.randn(2,3,4) y ``` ``` t…

TransE模型 PyTorch版本实现

版本1 === 出自 ``` import numpy as np import torch import torch.nn as nn class TransE(nn.Module): def __in…