FloatTensor x_i = tensor(x[:, None, :]) # (M, 1, 2) torch tensor y_j ... Operations on the last “vector” dimension (-1) or on optional “batch” dimensions ...
確定! 回上一頁