test_scale_sub_region_layer.py 315 字节
Newer Older
Y
yangyaming 已提交
1 2 3 4 5 6 7 8 9 10 11
from paddle.trainer_config_helpers import *

settings(batch_size=1000, learning_rate=1e-5)

data = data_layer(name='data', size=2016, height=48, width=42)
indices = data_layer(name='indices', size=6)

scale_sub_region = scale_sub_region_layer(
    input=data, indices=indices, value=0.0)

outputs(scale_sub_region)