torch.clone
时间: 2023-10-22 12:20:28 浏览: 163
PyTorch中张量的创建方法
`torch.clone` 是 PyTorch 中的一个函数,用于创建一个给定张量的副本。它会返回一个新的张量,该张量与原始张量具有相同的大小、形状和数据类型,并且其中的元素值也相同。
`torch.clone` 的语法如下:
```python
torch.clone(input, memory_format=None)
```
其中,参数 `input` 是要克隆的张量,而参数 `memory_format` 是可选的,用于指定新张量的内存布局格式。
下面是一个例子:
```python
import torch
x = torch.tensor([[1, 2], [3, 4]])
y = torch.clone(x)
print(x)
print(y)
```
运行结果如下:
```
tensor([[1, 2],
[3, 4]])
tensor([[1, 2],
[3, 4]])
```
可以看到,`y` 是 `x` 的一个副本,它们具有相同的值、形状和数据类型。
阅读全文