VimPlug/typeshed
1
0
Fork 0
mirror of https://github.com/davidhalter/typeshed.git synced 2026-05-04 20:45:49 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

[tensorflow]: Add tf.split (#14890)

Browse Source
This commit is contained in:
Hoël Bagard
2026-01-12 22:45:31 +09:00
committed by GitHub
parent 5c5f3eb390
commit 31791106cb
1 changed files with 7 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
stubs/tensorflow/tensorflow/__init__.pyi
+7
View File
@@ -385,6 +385,13 @@ def squeeze(
) -> Tensor: ...
@overload
def squeeze(input: RaggedTensor, axis: int | tuple[int, ...] | list[int], name: str | None = None) -> RaggedTensor: ...
def split(
value: TensorCompatible,
num_or_size_splits: int | TensorCompatible,
axis: int | Tensor = 0,
num: int | None = None,
name: str | None = "split",
) -> list[Tensor]: ...
def tensor_scatter_nd_update(
tensor: TensorCompatible, indices: TensorCompatible, updates: TensorCompatible, name: str | None = None
) -> Tensor: ...