blob: cc90ad333b2ffacb0d3dec247fe7963679233c5a (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
|
op {
graph_op_name: "SparseSplit"
in_arg {
name: "split_dim"
description: <<END
0-D. The dimension along which to split. Must be in the range
`[0, rank(shape))`.
END
}
in_arg {
name: "indices"
description: <<END
2-D tensor represents the indices of the sparse tensor.
END
}
in_arg {
name: "values"
description: <<END
1-D tensor represents the values of the sparse tensor.
END
}
in_arg {
name: "shape"
description: <<END
1-D. tensor represents the shape of the sparse tensor.
output indices: A list of 1-D tensors represents the indices of the output
sparse tensors.
END
}
out_arg {
name: "output_values"
description: <<END
A list of 1-D tensors represents the values of the output sparse
tensors.
END
}
out_arg {
name: "output_shape"
description: <<END
A list of 1-D tensors represents the shape of the output sparse
tensors.
END
}
attr {
name: "num_split"
description: <<END
The number of ways to split.
END
}
summary: "Split a `SparseTensor` into `num_split` tensors along one dimension."
description: <<END
If the `shape[split_dim]` is not an integer multiple of `num_split`. Slices
`[0 : shape[split_dim] % num_split]` gets one extra dimension.
For example, if `split_dim = 1` and `num_split = 2` and the input is
input_tensor = shape = [2, 7]
[ a d e ]
[b c ]
Graphically the output tensors are:
output_tensor[0] = shape = [2, 4]
[ a ]
[b c ]
output_tensor[1] = shape = [2, 3]
[ d e ]
[ ]
END
}
|