reshape

はじめに PyTorchには、テンソルを変形するメソッドとしてtorch.Tensor.viewとtorch.Tensor.reshapeが用意されています。本記事では、メソッドviewとreshapeの違いについてまとめます。 本記事のサマリ viewは要素が順に並んでいるときしか使えない。reshapeは、要素が順に並んでいないときでも、テンソルを変形できる テンソルの要素がメモリ上で順に並んでいるとは、テンソルの要素が連続したメモリに配置されているということ テンソルを転置すると、テンソルの要素がメモリ上で順に並んだ状態ではなくなり、viewを使えなくなる viewとreshapeの違い…