Split Forward Layer

The forward split layer copies a tensor from the input argument XR n 1 x n 2 x ... x n p to multiple output tensors.

Problem Statement

Given the number of output tensors k and a p-dimensional tensor X = (x i 1...i p ) of size n 1 x n 2 x ... x n p , the problem is to generate k p-dimensional tensors Y (j) (j=1, ..., k) of size n 1 x n 2 x ... x n p , such that:

For more complete information about compiler optimizations, see our Optimization Notice.
Select sticky button color: 
Orange (only for download buttons)