事前にリストを設定せずにfor文の内部で結合したい

Question

前提・実現したいこと

torchを使用しています。
for文で掛け算を行い，その結果を保存したいと考えています。

ここに質問の内容を詳しく書いてください。
下記ソースコードの部分にて，tensor型のinputsとweightsの掛け算を行い，
その結果をfor毎にinput_listとoutput_listに保存したいと考えています。

そのためにまず，input_list, output_listを定義するのですが，
（下記のコードでコメントアウトしてある部分です。）
このような書き方だと，input_list, output_listの出力の先頭にtorch.emptyが残ってしまったり，
または，tensor(0)には結合出来ませんとなってしまいます。

そこでリスト内包表記のように，事前にinput_list, output_listを作成せず，
for文を回せば良いかと思ったのですが，その場合だとtorch.matmulの掛け算をどのように入れ込むのか分かりませんでした。

input_listの先頭に余分な値を残さずに結合していくためにはどのようにする方法がありますでしょうか。
よろしくお願いします。

該当のソースコード

    def training(self,inputs,weights,steps):
        inputs = torch.rand(1,5)
        weights = torch.rand(5,5)

        # input_list=torch.empty(1,5)
        # output_list=torch.empty(1,5)
        
        # input_list=torch.tensor(0)
        # output_list=torch.tensor(0)

        for i in range(steps):
            outputs = torch.matmul(inputs,weights)
            
            input_list = torch.cat([input_list,inputs], axis=0)
            output_list = torch.cat([output_list,outputs], axis=0)

            inputs = output

        return inputs,input_list,output_list

試したこと

ここに問題に対して試したことを記載してください。

補足情報（FW/ツールのバージョンなど）

Google colaboratoryを使用しています。

Answer 1

初期値を以下にしてみてはどうでしょうか。
この作り方は、ググってもあまり見かけないのですが、forループの最初のtourch.catに対して何も無いように振舞うには、この形がよいと思いました。

        input_list=torch.tensor([])
        output_list=torch.tensor([])

Comments

[hiro_te]

問題なく完璧に動作しました。
ありがとうございました。