benchmarks.wiki

benchmarks.wiki / Public workspace

DS-1000 / 688 / 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 embeddings tensor A and B, which looks like
[
  [1,1,1],
  [1,1,1]
]


and 
[
  [0,0,0],
  [1,1,1]
]


what I want to do is calculate the L2 distance d(A,B) element-wise. 
First I did a tf.square(tf.sub(lhs, rhs)) to get
[
  [1,1,1],
  [0,0,0]
]


and then I want to do an element-wise reduce which returns 
[
  3,
  0
]


but tf.reduce_sum does not allow my to reduce by row. Any inputs would be appreciated. Thanks.


A:
<code>
import tensorflow as tf


a = tf.constant([
  [1,1,1],
  [1,1,1]
])
b = tf.constant([
  [0,0,0],
  [1,1,1]
])
</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
688
task_id
dbbf2bb2-cd6f-5f48-9a60-cb774ba002e0
task_key
default--test--688
task_revision_id
1
config / split
default / test

Official source

initial import