PyTorchのreshape関数の使い方は?

PyTorchでは、テンソルの形状を変更するためにreshape関数を使用する方法が2つあります。

  1. . 表示()
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]])
  1. リシェイプ()
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()メソッドは常に共有ストレージを持たない新しいテンソルを返します。

bannerAds