在一维TensorFlow张量中省略零,可以通过使用tf.boolean_mask函数来实现。tf.boolean_mask函数可以根据给定的掩码条件,从张量中选择满足条件的元素。
具体步骤如下:
这样,result就是省略了零的一维张量。
tf.boolean_mask函数的参数说明:
优势:
应用场景:
推荐的腾讯云相关产品和产品介绍链接地址:
领取专属 10元无门槛券
手把手带您无忧上云