|
|
@@ -54,5 +54,23 @@ namespace Tensorflow |
|
|
|
new_axis_mask: new_axis_mask, |
|
|
|
shrink_axis_mask: shrink_axis_mask, |
|
|
|
name: name); |
|
|
|
|
|
|
|
/// <summary> |
|
|
|
/// Splits a tensor into sub tensors. |
|
|
|
/// </summary> |
|
|
|
/// <param name="value">The Tensor to split.</param> |
|
|
|
/// <param name="num_split">Either an integer indicating the number of splits along split_dim or a 1-D integer |
|
|
|
/// Tensor or Python list containing the sizes of each output tensor along split_dim. |
|
|
|
/// If a scalar then it must evenly divide value.shape[axis]; otherwise the sum of sizes along the split dimension must match that of the value.</param> |
|
|
|
/// <param name="axis">An integer or scalar int32 Tensor. The dimension along which to split. Must be in the range [-rank(value), rank(value)). Defaults to 0.</param> |
|
|
|
/// <param name="name">A name for the operation (optional)</param> |
|
|
|
/// <returns>if num_or_size_splits is a scalar returns num_or_size_splits Tensor objects; |
|
|
|
/// if num_or_size_splits is a 1-D Tensor returns num_or_size_splits.get_shape[0] Tensor objects resulting from splitting value.</returns> |
|
|
|
public Tensor[] split(Tensor value, int num_split, Tensor axis, string name = null) => gen_array_ops.split( |
|
|
|
value: value, |
|
|
|
axis: axis, |
|
|
|
num_split: num_split, |
|
|
|
name: name |
|
|
|
); |
|
|
|
} |
|
|
|
} |