# ds1000 / 694 - taskset: [ds1000](https://harnessreport.com/tasks/ds1000.md) - difficulty: - category: - language: - runnable from the site: no - agent timeout: 1800s ## Results by harness _none yet_ ## Instruction ``` # 694: DS-1000 Task ## 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> ## What to do - Edit `solution/solution.py` so the code passes the DS-1000 tests. - Do not access the internet or install new packages; required libraries are preinstalled in the Docker image. - Run tests locally via `bash tests/test.sh`. ## Notes - Keep the variable names/signatures implied by the prompt/code_context. - The evaluator uses the original DS-1000 `code_context` (`test_execution` / `test_string`). ``` --- Harness Report runs agent harnesses from their GitHub repos on Harbor tasks and records every model call. Every page is also `.md` and `.json`; index: https://harnessreport.com/llms.txt · MCP: https://harnessreport.com/mcp