生成一个动态大小的range数组
huangguanMayday 发布于2019-12-29 浏览:602 回复:7
1
收藏
最后编辑于2021-05-17

正常的生成固定大小的range 数组的方法:
labels = fluid.layers.range(0, crmm_config.batch_size, 1, 'int64')

如果想生成动态大小的,如下代码执行有错(其中logits是n*n大小的二维数组,n是动态的):
shape_logits = fluid.layers.shape(logits)
labels = fluid.layers.range(0, shape_logits[0], 1, 'int64')

请问具体怎么用呢?

收藏
点赞
1
个赞
共7条回复 最后由用户已被禁言回复于2021-05-17
#8用户已被禁言回复于2021-05-17

mark

0
#3shippingwang回复于2019-12-29

shape_logits[0] 这块操作报的错么?用一下shape之后用slice取?

0
快速回复
TOP
切换版块