np.einsum('...j,...j->...', vf, vf)[-1, -1, -1]
np.einsum('...j,...j->...', vf, vf).dtype
np.einsum('...j,...j->...', vf, vf, dtype=np.int64)[-1, -1, -1]
самый быстрый способ найти величину (длину) квадрата векторного поля