Skip to content

Commit d488181

Browse files
authored
add back triu in fluid (#27135) (#27188)
1 parent 409eff7 commit d488181

File tree

2 files changed

+19
-1
lines changed

2 files changed

+19
-1
lines changed

python/paddle/fluid/layers/tensor.py

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -36,7 +36,7 @@
3636
'tensor_array_to_tensor', 'concat', 'sums', 'assign',
3737
'fill_constant_batch_size_like', 'fill_constant', 'argmin', 'argmax',
3838
'argsort', 'ones', 'zeros', 'reverse', 'has_inf', 'has_nan', 'isfinite',
39-
'range', 'linspace', 'zeros_like', 'ones_like', 'diag', 'eye'
39+
'range', 'linspace', 'zeros_like', 'ones_like', 'diag', 'eye', 'triu'
4040
]
4141

4242

@@ -1725,3 +1725,9 @@ def ones_like(x, out=None):
17251725
attrs={'value': 1.0},
17261726
outputs={'Out': [out]})
17271727
return out
1728+
1729+
1730+
@deprecated(since="2.0.0", update_to="paddle.triu")
1731+
def triu(input, diagonal=0, name=None):
1732+
import paddle
1733+
return paddle.tensor.triu(x=input, diagonal=diagonal, name=name)

python/paddle/fluid/tests/unittests/test_tril_triu_op.py

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -142,6 +142,18 @@ def test_api_with_dygraph(self):
142142
self.assertTrue(np.allclose(tril_out, np.tril(data)))
143143
self.assertTrue(np.allclose(triu_out, np.triu(data)))
144144

145+
def test_fluid_api(self):
146+
data = np.random.random([1, 9, 9, 4]).astype('float32')
147+
x = fluid.data(shape=[1, 9, -1, 4], dtype='float32', name='x')
148+
triu_out = fluid.layers.triu(x)
149+
150+
place = fluid.CUDAPlace(0) if fluid.core.is_compiled_with_cuda(
151+
) else fluid.CPUPlace()
152+
exe = fluid.Executor(place)
153+
triu_out = exe.run(fluid.default_main_program(),
154+
feed={"x": data},
155+
fetch_list=[triu_out])
156+
145157

146158
if __name__ == '__main__':
147159
unittest.main()

0 commit comments

Comments
 (0)