PyTorchのreshape関数の使い方は?
PyTorchでは、テンソルの形状を変更するためにreshape関数を使用する方法が2つあります。
- . 表示()
import torch
# 创建一个大小为(2, 3)的张量
x = torch.tensor([[1, 2, 3], [4, 5, 6]])
# 使用view方法将张量的形状改变为(3, 2)
y = x.view(3, 2)
print(y)
情報を日本語で自然な形で言い換えてください。一つのオプションで良いです。
答え: 私たちは新しいプロジェクトの計画に取り組んでいます。
tensor([[1, 2],
[3, 4],
[5, 6]])
- リシェイプ()
import torch
# 创建一个大小为(2, 3)的张量
x = torch.tensor([[1, 2, 3], [4, 5, 6]])
# 使用reshape方法将张量的形状改变为(3, 2)
y = x.reshape(3, 2)
print(y)
出力:
tensor([[1, 2],
[3, 4],
[5, 6]])
これらの2つの方法は、テンソルの形状を変更するために使用できますが、.view()メソッドは場合によっては共有ストレージのビューを返す可能性があり、.reshape()メソッドは常に共有ストレージを持たない新しいテンソルを返します。