wehub-resource-sync
20 lines
481 B
Python
Executable File
20 lines
481 B
Python
Executable File
from prml.nn.array.broadcast import broadcast_to
|
|
from prml.nn.array.flatten import flatten
|
|
from prml.nn.array.reshape import reshape, reshape_method
|
|
from prml.nn.array.split import split
|
|
from prml.nn.array.transpose import transpose, transpose_method
|
|
from prml.nn.tensor.tensor import Tensor
|
|
|
|
|
|
Tensor.flatten = flatten
|
|
Tensor.reshape = reshape_method
|
|
Tensor.transpose = transpose_method
|
|
|
|
__all__ = [
|
|
"broadcast_to",
|
|
"flatten",
|
|
"reshape",
|
|
"split",
|
|
"transpose"
|
|
]
|