VOLANTIS 2.6.6

PyTorch Introduction

tensor张量

1、与numpy的相互转换

1
2
3
4
#tensor->numpy ：
y_numpy = torch.numpy(x_tensor)
#numpy->tensor ：
y_torch = torch.from_numpy(x_numpy)

2、tensor可以转变矩阵形状，只要总共的元素数不变就行

1
2
3
4
5
6
#tensor.view
x = torch.rand(5,3)
y = x.view(15)
z = x.view(-1, 5)  # -1所指的维度可以根据其他维度的值推出来
#只有一个维度可以赋值-1！
print(x.size(), y.size(), z.size())

3、


3 / 5