frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

Symbolic representation for unfold using ATen (#3334)

Browse Source
This commit is contained in:
James Reed
2017-10-27 21:08:45 -07:00
committed by Edward Z. Yang
parent ac8f56656d
commit 204044a522
1 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
torch/onnx/symbolic.py
View File

@ -233,3 +233,7 @@ def avg_pool2d(g, input, kernel_size, stride, padding, ceil_mode, count_include_
def log_softmax(g, input, dim=None):
return g.op("Log", g.op('Softmax', input, axis_i=dim).setTypeAs(input))
def unfold(g, input, dimension, size, step):
return g.op("ATen", input, operator_s="unfold", dimension_i=dimension, size_i=size, step_i=step)