Skip to content

Commit 23816af

Browse files
committed
fix Torch's docs
1 parent 558e33c commit 23816af

File tree

2 files changed

+2
-2
lines changed
  • docs/backend-reference
  • i18n/zh/docusaurus-plugin-content-docs/current/backend-reference

2 files changed

+2
-2
lines changed

docs/backend-reference/torch.mdx

+1-1
Original file line numberDiff line numberDiff line change
@@ -94,7 +94,7 @@ flowchart LR
9494

9595
## Torch
9696
Similar to [SyncTensor](#synctensor), but with additional cross-card functionality. Effective from version 0.3.2b1.
97-
- Torch::backend: required
97+
- Torch::backend: required, like `Torch[TensorrtTensor]`
9898
- device_id: Default is -1, which sets the current device to this ID. During initialization, it is equivalent to calling `c10::cuda::set_device(device_id)` or `torch.cuda.set_device(device_id)`. During forward propagation, the input data type must be `at::Tensor` or `vector<at::Tensor>`. This backend will move the data to the specified graphics card.
9999

100100
## SaveTensor

i18n/zh/docusaurus-plugin-content-docs/current/backend-reference/torch.mdx

+1-1
Original file line numberDiff line numberDiff line change
@@ -97,7 +97,7 @@ flowchart LR
9797

9898
## Torch
9999
[SyncTensor](#synctensor)类似,但额外添加了跨卡功能。 从0.3.2b1版本开始生效。
100-
- Torch::backend: default=Identity
100+
- Torch::backend: 必须提供, 比如 `Torch[TensorrtTensor]`
101101
- device_id: 默认为-1;表示将当前设备设置为这个id. 初始化时内部相当于调用了`c10::cuda::set_device(device_id)` 或者 `torch.cuda.set_device(device_id)`,前向时,要求输入数据类型是`at::Tensor` or `vector<at::Tensor>`; 此后端会将数据搬运到指定显卡上。
102102

103103
## SaveTensor

0 commit comments

Comments
 (0)