複数の寸法のテンソルを考えると、それが単一の次元を持つようにそれを平らにするのですか?
例えば:
>>> t = torch.Rand([2, 3, 5])
>>> t.shape
torch.Size([2, 3, 5])
_
どのようにして図形を持つように平らにしますか?
torch.Size([30])
_
あなたは簡単なt.view(-1)
をすることができます
>>>t = torch.Rand([2, 3, 5])
>>>t = t.view(-1)
>>>t.shape
torch.Size([30])
_