tf.dynamic_partition函数

本文深入探讨了使用TensorFlow进行矩阵操作的方法,包括如何将多维列表转换为一维张量,利用条件语句对元素进行筛选,并通过dynamic_partition函数按条件分割矩阵,最后在会话中运行并打印结果。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

将原矩阵按照索引矩阵的方式分割

alist=[[[[1,1,1],[2,2,2],[3,3,3]],[[4,4,4],[5,5,5],[6,6,6]]],[[[7,7,7],[8,8,8],[9,9,9]],[[10,10,10],[11,11,11],[12,12,12]]]]
aal=tf.reshape(alist, [-1, ])
t=tf.to_int32(aal <= 5)
valid_logits = tf.dynamic_partition(aal, t, num_partitions=2)
with tf.Session() as sess:
    print(sess.run(valid_logits))

输出:

[array([ 6,  6,  6,  7,  7,  7,  8,  8,  8,  9,  9,  9, 10, 10, 10, 11, 11,
       11, 12, 12, 12]), array([1, 1, 1, 2, 2, 2, 3, 3, 3, 4, 4, 4, 5, 5, 5])]
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值