Skip to content

GitLab

Unverified Commit 511aeb88 authored by Jiacheng Huang's avatar Jiacheng Huang Committed by GitHub
Browse files

issue/495 修复 `infinicore.Tensor.copy_` 和 `infinicore.Tensor.as_strided` 的行为错误

parent 93e7d887
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
python/infinicore/tensor.py
View file @ 511aeb88
...@@ -57,7 +57,7 @@ class Tensor: ...@@ -57,7 +57,7 @@ class Tensor:
return self._underlying.is_is_pinned() return self._underlying.is_is_pinned()
def copy_(self, src): def copy_(self, src):
return Tensor(self._underlying.copy_(src._underlying)) self._underlying.copy_(src._underlying)
def to(self, *args, **kwargs): def to(self, *args, **kwargs):
return Tensor( return Tensor(
...@@ -65,7 +65,7 @@ class Tensor: ...@@ -65,7 +65,7 @@ class Tensor:
) )
def as_strided(self, size, stride): def as_strided(self, size, stride):
Tensor(self._underlying.as_strided(size, stride)) return Tensor(self._underlying.as_strided(size, stride))
def contiguous(self): def contiguous(self):
return Tensor(self._underlying.contiguous()) return Tensor(self._underlying.contiguous())
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment