import tensorflow as tf
import numpy as np
a = np.zeros((4,2))
b = tf.random.categorical(a, 1)
x = b
y = tf.one_hot(x, depth=2)
print(tf.shape(y))
aW1wb3J0IHRlbnNvcmZsb3cgYXMgdGYKaW1wb3J0IG51bXB5IGFzIG5wCiAKYSA9IG5wLnplcm9zKCg0LDIpKQpiID0gdGYucmFuZG9tLmNhdGVnb3JpY2FsKGEsIDEpCnggPSBiCnkgPSB0Zi5vbmVfaG90KHgsIGRlcHRoPTIpCnByaW50KHRmLnNoYXBlKHkpKQ==