Function Split

tensorflow/go/op/wrappers.go:3171–3194

Splits a tensor into `num_split` tensors along one dimension. Arguments: axis: 0-D. The dimension along which to split. Must be in the range `[-rank(value), rank(value))`. value: The tensor to split. num_split: The number of ways to split. Must evenly divide `value.shape[split_dim]`. Returns

(scope *Scope, axis tf.Output, value tf.Output, num_split int64)

// except along `axis`, where their sizes are
// `values.shape[split_dim] / num_split`.
func Split(scope *Scope, axis tf.Output, value tf.Output, num_split int64) (output []tf.Output) {
	if scope.Err() != nil {
		return
	}
	attrs := map[string]interface{}{"num_split": num_split}
	opspec := tf.OpSpec{
		Type: "Split",
		Input: []tf.Input{
			axis, value,
		},
		Attrs: attrs,
	}
	op := scope.AddOperation(opspec)
	if scope.Err() != nil {
		return
	}
	var idx int
	var err error
	if output, idx, err = makeOutputList(op, idx, "output"); err != nil {
		scope.UpdateErr("Split", err)
		return
	}
	return output
}

// Computes offsets of concat inputs within its output.
//