tensorflow::ops

Function all_to_all

Source
pub fn all_to_all<O0: Into<Output>, O1: Into<Output>>(
    input: O0,
    group_assignment: O1,
    scope: &mut Scope,
) -> Result<Operation>
Expand description

Shorthand for AllToAll::new().build(input, group_assignment, scope).