как инициализировать Тензор для torch.cat

Question

import torch

#Y_pred = ?

for xi in X_iter:
    y_pred = net(xi).argmax(dim=1)
    Y_pred = torch.cat([Y_pred, y_pred])

Как вы инициализируете этот тензор, или есть лучший способ написать его?

Answer 1

Вы можете сделать это вместо:

Y_pred = torch.cat([net(xi).argmax(dim=1) for xi in X_iter])