Crate caffe2op_spacebatch

Structs

BatchToSpaceOp
| Rearranges (permutes) data from batch | into blocks of spatial data, followed | by cropping. | | This is the reverse transformation | of SpaceToBatch. | | More specifically, this op outputs | a copy of the input tensor where values | from the batch dimension are moved in | spatial blocks to the height and width | dimensions, followed by cropping along | the height and width dimensions. | | Only “NCHW” order is currently supported. | | Github Links: | | - https://github.com/pytorch/pytorch/blob/master/caffe2/operators/space_batch_op.cc |
GetBatchToSpaceGradient
GetSpaceToBatchGradient
SpaceBatchOpBase
SpaceToBatchOp
| Zero-pads and then rearranges (permutes) | blocks of spatial data into batch. More | specifically, this op outputs a copy | of the input tensor where values from | the height and width dimensions are | moved to the batch dimension. After | the zero-padding is according to the | pad argument, both height and width | of the input must be divisible by the | block_size. Only “NCHW” order is | currently supported. | | Github Links: | | - https://github.com/pytorch/pytorch/blob/master/caffe2/operators/space_batch_op.cc |