(d, x1, A, x2)
| 19 | |
| 20 | |
| 21 | def tensor_mul(d, x1, A, x2): |
| 22 | A = tf.reshape(A, [d, d*d]) |
| 23 | # (1 x d) x (d x dd) |
| 24 | tmp = tf.matmul(x1, A) |
| 25 | # (1 x dd) |
| 26 | tmp = tf.reshape(tmp, [d, d]) |
| 27 | # (d x d) |
| 28 | tmp = tf.matmul(tmp, tf.transpose(x2)) |
| 29 | # (d x 1) |
| 30 | return tf.reshape(tmp, [1, d]) |
| 31 | |
| 32 | |
| 33 | def get_labels(tree): |
no outgoing calls
no test coverage detected