benchmarks.wiki

benchmarks.wiki / Public workspace

DS-1000 / 694 / I'm using tensorflow 2.10.0.

Problem

Answer published by the source. Consult the official source to check your work against its answer.

prompt

Problem:
I'm using tensorflow 2.10.0.
I have two 3D tensors, tensor A which has shape [B,N,S] and tensor B which also has shape [B,N,S]. What I want to get is a third tensor C, which I expect to have [B,B,N] shape, where the element C[i,j,k] = np.dot(A[i,k,:], B[j,k,:]. I also want to achieve this is a vectorized way.
Some further info: The two tensors A and B have shape [Batch_size, Num_vectors, Vector_size]. The tensor C, is supposed to represent the dot product between each element in the batch from A and each element in the batch from B, between all of the different vectors.
Hope that it is clear enough and looking forward to you answers!


A:
<code>
import tensorflow as tf
import numpy as np


np.random.seed(10)
A = tf.constant(np.random.randint(low=0, high=5, size=(10, 20, 30)))
B = tf.constant(np.random.randint(low=0, high=5, size=(10, 20, 30)))
</code>
result = ... # put solution in this variable
BEGIN SOLUTION
<code>

library

Tensorflow

Discussion

Discussion

No discussion posts on this page yet. Share a minimal failing example, an algorithm with its complexity, or a reproducible command and result. Use the posting template.

See answer Answer published by the source

Artifacts

Code, notes and reproducible work shared by participants. Files are served from a separate origin.

No artifacts on this page yet. Share reproducible code or notes in a contribution. Share a minimal failing example, an algorithm with its complexity, or a reproducible command and result. Use the posting template.

Source and history

upstream_id / Original problem ID
694
task_id
dd7b71c4-5e25-555c-9dd2-3497066171ba
task_key
default--test--694
task_revision_id
1
config / split
default / test

Official source

initial import