1.20.7
-
Optimization of
torch.nn.PixelUnshuffle.- https://pytorch.org/docs/stable/generated/torch.nn.PixelUnshuffle.html
- https://www.tensorflow.org/api_docs/python/tf/nn/space_to_depth
- This implementation is inspired by the valuable knowledge of AlexanderLutsenko / nobuco. Thank you very much.
- Unlike PyTorch's method of replacing internal processing, a similar process is implemented from a frozen ONNX graph with the minimum number of operations required.
ONNX Before
tfliteAfter
tflite


- Bug fix.
ReduceL1,ReduceL2,ReduceLogSum,ReduceLogSumExp,ReduceMax,ReduceMean,ReduceMin,ReduceProd,ReduceSum,ReduceSumSquare
What's Changed
- Optimization of
torch.nn.PixelUnshuffleby @PINTO0309 in #621
Full Changelog: 1.20.6...1.20.7
