Struct MapBuilder 

pub struct MapBuilder { /* private fields */ }

Builder for Map constructs. Created by FlowBuilder::map.

Implementations

impl MapBuilder

pub fn batched(self) -> Self

Fast path: pass full tensor to body in one call instead of element-wise.

pub fn each(self) -> FlowBuilder

Apply body to each element along dim 0 of the current stream.

pub fn over(self, tag: &str) -> FlowBuilder

Iterate over a tagged tensor (backward ref) instead of current stream.

pub fn slices(self, n: i64) -> FlowBuilder

Decompose last dim into n slices, map body over each, recompose.

For input [B, D] with slices(n): reshape [B, D] → [B*n, D/n], map body over B*n elements, reshape back to [B, outD*n]. D must be divisible by n.