Skip to content

Commit e29b7c4

Browse files
committed
add tf.range
1 parent 56cbde4 commit e29b7c4

1 file changed

Lines changed: 12 additions & 0 deletions

File tree

stubs/tensorflow/tensorflow/__init__.pyi

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -467,4 +467,16 @@ def clip_by_value(
467467
t: Tensor | IndexedSlices, clip_value_min: TensorCompatible, clip_value_max: TensorCompatible, name: str | None = None
468468
) -> Tensor: ...
469469
def tile(input: RaggedTensorLike, multiples: Tensor | Sequence[int], name: str | None = None) -> Tensor: ...
470+
@overload
471+
def range(
472+
limit: int | Tensor, /, *, delta: int | Tensor = 1, dtype: DTypeLike | None = None, name: str | None = "range"
473+
) -> Tensor: ...
474+
@overload
475+
def range(
476+
start: int | Tensor = 0,
477+
limit: int | Tensor = 0,
478+
delta: int | Tensor = 1,
479+
dtype: DTypeLike | None = None,
480+
name: str | None = "range",
481+
) -> Tensor: ...
470482
def __getattr__(name: str): ... # incomplete module

0 commit comments

Comments
 (0)