|
|
@@ -36,7 +36,6 @@ __all__ = [ |
|
|
|
"broadcast", |
|
|
|
"concat", |
|
|
|
"cond_take", |
|
|
|
"dimshuffle", |
|
|
|
"expand_dims", |
|
|
|
"eye", |
|
|
|
"flatten", |
|
|
@@ -730,12 +729,6 @@ def transpose(inp: Tensor, pattern: Iterable[int]) -> Tensor: |
|
|
|
return inp.transpose(pattern) |
|
|
|
|
|
|
|
|
|
|
|
def dimshuffle(inp: Tensor, pattern: Iterable[int]) -> Tensor: |
|
|
|
r"""Same as :func:`~.transpose`. |
|
|
|
""" |
|
|
|
return transpose(inp, pattern) |
|
|
|
|
|
|
|
|
|
|
|
def reshape(inp: Tensor, target_shape: Iterable[int]) -> Tensor: |
|
|
|
r""" |
|
|
|
Reshapes a tensor to given target shape; total number of logical elements must |
|
|
|