code로 이해하는 SR-GNN 논문
As 행렬 만들기 v1 -> v2 -> v3 -> v2 -> v4의 session이 존재할때 그래프는 위의 노드와 간선 연결이 보인다. 해당 session의 그래프를 바탕으로 connectionm matrix A_s 생성 A_s는 두개의 인접행렬 A_s(out)과 A_s(in)의 연결로 정의된다. 이는 즉, 세션 그래프에서 각각 진입, 진출의 간선 연결로 생각하면 된다. # batch가 10이라고 가정 i = [0, 1, 2, 3, 4, 5, 6, 7, 8, 9] # input data, masking된 데이터, target 데이터 불러오기 inputs, mask, targets = train_data.inputs[i], train_data.mask[i], train_data.targets[i] item..
2021. 5. 21.
Tensorflow vs Pytorch -(4)
tf.transpose(a) x = tf.constant([[1, 2, 3], [4, 5, 6]]) x2 = tf.transpose(x) with tf.Session() as sess: print(sess.run(x)) print(sess.run(x2)) [[1, 2, 3], [4, 5, 6]] [[1 4] [2 5] [3 6]] torch.transpose(input, dim0, dim1) x = torch.tensor([[1, 2, 3], [4, 5, 6]]) torch.transpose(x, 0, 1) tensor([[1, 2, 3], [4, 5, 6]]) tensor([[1, 4], [2, 5], [3, 6]]) dim0과 dim1을 swap한다. Loss function # mean square..
2021. 5. 19.