Skip to content
Projects
Groups
Snippets
Help
Loading...
Sign in / Register
Toggle navigation
N
ngraph
Project
Project
Details
Activity
Cycle Analytics
Repository
Repository
Files
Commits
Branches
Tags
Contributors
Graph
Compare
Charts
Issues
0
Issues
0
List
Board
Labels
Milestones
Merge Requests
0
Merge Requests
0
CI / CD
CI / CD
Pipelines
Jobs
Schedules
Charts
Packages
Packages
Wiki
Wiki
Snippets
Snippets
Members
Members
Collapse sidebar
Close sidebar
Activity
Graph
Charts
Create a new issue
Jobs
Commits
Issue Boards
Open sidebar
submodule
ngraph
Commits
768a8e13
Commit
768a8e13
authored
Feb 11, 2020
by
mbencer
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
Update doc
parent
cf713136
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
7 additions
and
2 deletions
+7
-2
split.hpp
src/ngraph/builder/split.hpp
+7
-2
No files found.
src/ngraph/builder/split.hpp
View file @
768a8e13
...
@@ -58,9 +58,13 @@ namespace ngraph
...
@@ -58,9 +58,13 @@ namespace ngraph
/// \param[in] value The value to be split.
/// \param[in] value The value to be split.
/// \param[in] length_parts The vector defining the lengths of each split part.
/// \param[in] length_parts The vector defining the lengths of each split part.
/// \param[in] axis The axis we split input node on. Default value is zero
/// \param[in] axis The axis we split input node on. Default value is zero
/// axis.
/// axis.
/// \note This implementation supports negative `axis` values (similar to NumPy
/// indexing). This means that the axis to split on will be counted from
/// the back of the tensor (negative values are subtracted from its rank).
///
///
/// \return The vector containing multiple nodes we split input node into.
/// \return The vector containing multiple nodes we split input node into.
/// The vector is output of Split:v1 op
///
///
NodeVector
split
(
const
Output
<
Node
>&
value
,
NodeVector
split
(
const
Output
<
Node
>&
value
,
const
std
::
vector
<
size_t
>&
length_parts
,
const
std
::
vector
<
size_t
>&
length_parts
,
...
@@ -73,13 +77,14 @@ namespace ngraph
...
@@ -73,13 +77,14 @@ namespace ngraph
/// axis. The length of the axis to split must be divisible by
/// axis. The length of the axis to split must be divisible by
/// this value.
/// this value.
/// \param[in] axis The axis we split input node on. Default value is zero
/// \param[in] axis The axis we split input node on. Default value is zero
/// axis.
///
axis.
///
///
/// \note This implementation supports negative `axis` values (similar to NumPy
/// \note This implementation supports negative `axis` values (similar to NumPy
/// indexing). This means that the axis to split on will be counted from
/// indexing). This means that the axis to split on will be counted from
/// the back of the tensor (negative values are subtracted from its rank).
/// the back of the tensor (negative values are subtracted from its rank).
///
///
/// \return The vector containing multiple nodes we split input node into.
/// \return The vector containing multiple nodes we split input node into.
/// The vector is output of VariadicSplit:v1 op
///
///
NodeVector
split
(
const
Output
<
Node
>&
value
,
size_t
split_parts
,
int64_t
axis
=
0
);
NodeVector
split
(
const
Output
<
Node
>&
value
,
size_t
split_parts
,
int64_t
axis
=
0
);
}
}
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment