oneflow.advanced¶
Advanced features¶
-
oneflow.advanced.
distribute_add
(xs: Sequence[oneflow._oneflow_internal.BlobDesc], name: Optional[str] = None) → oneflow._oneflow_internal.BlobDesc¶
-
oneflow.advanced.
distribute_clone
(x: oneflow._oneflow_internal.BlobDesc, name: Optional[str] = None) → Tuple[oneflow._oneflow_internal.BlobDesc]¶
-
oneflow.advanced.
distribute_concat
(xs: Sequence[oneflow._oneflow_internal.BlobDesc], axis: int = 0, name: Optional[str] = None) → oneflow._oneflow_internal.BlobDesc¶
-
oneflow.advanced.
distribute_map
(xs: Union[Sequence[oneflow._oneflow_internal.BlobDesc], oneflow._oneflow_internal.BlobDesc], f: Callable[[oneflow._oneflow_internal.BlobDesc, oneflow._oneflow_internal.BlobDesc], oneflow._oneflow_internal.BlobDesc], axis: int = 0) → Tuple[oneflow._oneflow_internal.BlobDesc]¶
-
oneflow.advanced.
distribute_split
(x: oneflow._oneflow_internal.BlobDesc, axis: int = 0, name: Optional[str] = None) → Tuple[oneflow._oneflow_internal.BlobDesc]¶