Python wrappers
Public Python APIs implemented by this engine symbol.Split a tensor into a fixed number of equal chunks along an axis.
Alias for split_op with num_splits = chunks. Provided for
API symmetry with reference framework's chunk.
See Also
split_op.
Parameters
aTensorImplPtrInput tensor.
chunksint64_tNumber of chunks; must divide
a.shape[axis] exactly.axisintAxis to chunk along.
Returns
vector<TensorImplPtr>chunks equal-sized tensors.