znsoft的专栏

from winmltools.utils import convert_float_to_float16from winmltools.utils import load_model, save_modelonnx_model = load_model('model.onnx')new_onnx_model = convert_float_to_float16(onnx_model)save_model(new_onnx_model, 'model_fp16.onnx')

一段代码说明：import torchimport torch.nn.functional as Fdata=torch.randn(2,5)softmax= F.softmax(data,dim=1)log_soft1=torch.log(log)log_soft=F.log_softmax(data,dim=1) # softmax + log, 即先softmax再log#上面的log_soft 和log_soft1 是等值 的target=torch.tensor.

点乘相应点相乘，x.mul(y) ，即点乘操作，点乘不求和操作，又可以叫作Hadamard productimport torchx = torch.tensor([[3,3],[3,3]])y = x*x #x.dot(x)z = torch.mul(x,x) #x.mul(x)print(y)print(z)显示结果：tensor([[9, 9], [9, ...

新的pytorch 已经发布，在conda下安装1.0的命令如下： 支持cuda 10.0 版本 conda install pytorch torchvision cudatoolkit=10.0 -c pytorch