PyTorch Forums
h_j