Tensorflow2.x:tensor切片


目标

利用Tensorflow 2.x API 实现以下两个目标:

  1. 对张量(tensor)切片。
  2. 在张量(tensor)的指定索引处插入数据。

提取张量切片

一维张量

t1 = tf.constant([0,1,2,3,4,5,6,7])

tf.slice(t1, begin=[1], size=[3])  # 利用tf.slice()切片    
t1[0:4]                            # 利用索引切片
t1[-3:]

高维张量

t2 = tf.constant([[0,1,2,3,4],
                 [5,6,7,8,9],
                 [10,11,12,13,14],
                 [15,16,17,18,19]])

tf.slice(t2, begin=[1,1], size=[1,2])
t2[:-1, 0:3]

tf.strided_slice :通过设定步长来提取张量切片

tf.gather :从张量的单个轴中提取特定索引

# 两种方式等价
tf.gather(t1,indices=[0,3,6]) 
t1[::3]

字符串tensor也适用:

alphabet = tf.constant(list
  • 0
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值