Skip to content

Commit

Permalink
add Tensor
Browse files Browse the repository at this point in the history
  • Loading branch information
Asthestarsfalll committed Apr 24, 2024
1 parent 8121612 commit dd9e04e
Showing 1 changed file with 22 additions and 0 deletions.
22 changes: 22 additions & 0 deletions python/paddle/__init__.pyi
Original file line number Diff line number Diff line change
@@ -1,5 +1,27 @@
from __future__ import annotations

from .framework.dtype import (
bfloat16 as bfloat16,
bool as bool,
complex64 as complex64,
complex128 as complex128,
dtype as dtype,
finfo as finfo,
float16 as float16,
float32 as float32,
float64 as float64,
iinfo as iinfo,
int8 as int8,
int16 as int16,
int32 as int32,
int64 as int64,
uint8 as uint8,
)

class Tensor: ...

def to_tensor(data, dtype=None, place=None, stop_gradient=True) -> Tensor: ...

class CPUPlace: ...

class CUDAPlace:
Expand Down

0 comments on commit dd9e04e

Please sign in to comment.