Tensor computation about blog HOT 1 OPEN

import alpa.numpy as ap
import numpy as np

tensor = np.random.rand(1000000, 100)
t_transposed = ap.transpose(tensor)
dot_matrix = ap.dot(
    tensor, t_transposed)
v1_row_norm = ap.linalg.norm(
    tensor, axis=1).reshape(-1, 1)
v2_col_norm = ap.linalg.norm(
    t_transposed, axis=0).reshape(1, -1)
norm_matrix = ap.dot(
    v1_row_norm, v2_col_norm)
res = dot_matrix / norm_matrix
res = ap.where(ap.isneginf(res), 0, res)
res.block_until_ready()