Browse Source

docs(mge/functional): update copy example

GitOrigin-RevId: 2ea102e0ad
release-1.11.1
Megvii Engine Team 2 years ago
parent
commit
6e48e59334
1 changed files with 2 additions and 1 deletions
  1. +2
    -1
      imperative/python/megengine/functional/tensor.py

+ 2
- 1
imperative/python/megengine/functional/tensor.py View File

@@ -1050,7 +1050,8 @@ def copy(inp, device=None):


Examples: Examples:


>>> x = Tensor([1, 2, 3], numpy.int32)
>>> import numpy as np
>>> x = Tensor([1, 2, 3], np.int32)
Tensor([1 2 3], dtype=int32, device=xpux:0) Tensor([1 2 3], dtype=int32, device=xpux:0)


>>> F.copy(x, 'cpu1') >>> F.copy(x, 'cpu1')


Loading…
Cancel
Save