Skip to content

Commit 410287e

Browse files
committed
add tf.stack
1 parent a971b2a commit 410287e

1 file changed

Lines changed: 1 addition & 0 deletions

File tree

stubs/tensorflow/tensorflow/__init__.pyi

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -410,6 +410,7 @@ def split(
410410
num: int | None = None,
411411
name: str | None = "split",
412412
) -> list[Tensor]: ...
413+
def stack(values: TensorCompatible, axis: int = 0, name: str | None = "stack") -> Tensor: ...
413414
def tensor_scatter_nd_update(
414415
tensor: TensorCompatible, indices: TensorCompatible, updates: TensorCompatible, name: str | None = None
415416
) -> Tensor: ...

0 commit comments

Comments
 (0)