Index assignment to tensor objects

Question

I have four tensor objects, e.g., A, B, C, and D (see below).

A: <tf.Tensor 'mul_388:0' shape=(5,) dtype=float32>
B: <tf.Tensor 'mul_396:0' shape=(5,) dtype=float32>
C: <tf.Tensor 'clip_by_value_30:0' shape=(5,) dtype=float32>
D: <tf.Tensor 'clip_by_value_31:0' shape=(5,) dtype=float32>

I have to create new tensor object E based on the conditions given below:

E = C when A <= B
else
E = D

What is the best way to do it? I am using Tensorflow version 2.9.1.

I tried following but didn't work.

E = C[A <= B]
E = D[A > B]

AndrzejO · Accepted Answer · 2022-10-02 02:40:51Z

1

Use tf.cond:

E = tf.cond(A <= B, true_fn=lambda: C, false_fn=lambda: D)

answered Oct 2, 2022 at 2:40

AndrzejO

1,5721 gold badge10 silver badges15 bronze badges

Sign up to request clarification or add additional context in comments.

2 Comments

elbe Over a year ago

Can also do E = tf.where(A <= B, C, D)

raghav Over a year ago

For me, E = tf.where(A <= B, C, D) worked fine.

Collectives™ on Stack Overflow

Index assignment to tensor objects

1 Answer 1

2 Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

2 Comments

Your Answer

Sign up or log in

Post as a guest

Related