Merge branch 'master' into liqun/onnx17Stage2
This commit is contained in:
Коммит
b09ff2bd90
|
@ -13,9 +13,25 @@
|
|||
- ``gather`` operation now supports an axis argument
|
||||
- ``squeeze`` and ``expand_dims`` operations for easily removing and adding singleton axes
|
||||
- ``zeros_like`` and ``ones_like`` operations. In many situations you can just rely on CNTK correctly broadcasting a simple 0 or 1 but sometimes you need the actual tensor.
|
||||
- ``depth_to_space``: Rearranges elements in the input tensor from the depth dimension into spatial blocks. Typical use of this operation is for implementing sub-pixel convolution for some image super-resolution models.
|
||||
- ``space_to_depth``: Rearranges elements in the input tensor from the spatial dimensions to the depth dimension. It is largely the inverse of DepthToSpace.
|
||||
|
||||
## ONNX
|
||||
- Improved ONNX support in CNTK.
|
||||
- Update ONNX to the latest ONNX from https://github.com/onnx/onnx
|
||||
- Fixed several bugs.
|
||||
There have been several improvements to ONNX support in CNTK.
|
||||
|
||||
### Updates
|
||||
- Updated ONNX ``Reshape`` op to handle ``InferredDimension``.
|
||||
- Adding ``producer_name`` and ``producer_version`` fields to ONNX models.
|
||||
- Handling the case when neither ``auto_pad`` nor ``pads`` atrribute is specified in ONNX ``Conv`` op.
|
||||
|
||||
### Bug fixes
|
||||
- Fixed bug in ONNX ``Pooling`` op serialization
|
||||
- Bug fix to create ONNX ``InputVariable`` with only one batch axis.
|
||||
- Bug fixes and updates to implementation of ONNX ``Transpose`` op to match updated spec.
|
||||
- Bug fixes and updates to implementation of ONNX ``Conv``, ``ConvTranspose``, and ``Pooling`` ops to match updated spec.
|
||||
|
||||
## Operators
|
||||
### Group convolution
|
||||
- Fixed bug in group convolution. Output of CNTK ``Convolution`` op will change for groups > 1. More optimized implementation of group convolution is expected in the next release.
|
||||
- Better error reporting for group convolution in ``Convolution`` layer.
|
||||
|
||||
|
|
Загрузка…
Ссылка в новой задаче