/usr/local/lib64/python3.6/site-packages/caffe2/contrib/aten/docs/__pycache__
Edit: /usr/local/lib64/python3.6/site-packages/caffe2/contrib/aten/docs/__pycache__/sample.cpython-36.pyc (1647B)
3
Eg @ s@ d dl Z d dlZd dlmZ d dlmZmZ d dlZd dl Z d dl
Z G dd deZ G dd dej
Ze j Zej je eejdd eejdd fed
d e jejZejjdd jejZejjdd jejZe j j j j eZ ejj d jeejj d jeiZ!e j"e!d Z#ej$ed Z%e%e% e Z&ej'j(e&e# dS )
N)nn)VariableFunctionc @ s$ e Zd Zedd Zedd ZdS )
MyFunctionc C s || | S )N )ctxxyr r K/usr/local/lib64/python3.6/site-packages/caffe2/contrib/aten/docs/sample.pyforward
s zMyFunction.forwardc C s | j d||}| j d||}|S )Nmuladd)Zat)graphr r Zx2rr r r
symbolic s zMyFunction.symbolicN)__name__
__module__ __qualname__ staticmethodr r r r r r
r s r c @ s e Zd Zdd ZdS )MyModulec C s t j |}tj||S )N)r ZReLUr apply)selfr r r r r
r s zMyModule.forwardN)r r r r r r r r
r s r T)verbose ))tempfileZnumpynpZtorchr Ztorch.autogradr r Z
torch.onnxZonnxZcaffe2.python.onnx.backendZcaffe2r Moduler NamedTemporaryFilefZexportZonesloadnamer randomZrandnZastypeZfloat32abpythonbackendprepareZprepared_backendinputWrunZc2_outmaximumr r ZtestingZassert_array_almost_equalr r r r
s*