Skip to content

ExpandLayer get wrong result #5013

Closed
@wanghaoshuang

Description

@wanghaoshuang
input(x):
seq = [a, b], [c]
seq_starts = [0, 2, 3]

input(y):
seq_starts = [0, 2, 3, 5]
sub_seq_starts = [0, 3, 5]

output = ?

我按原paddle expand layer的计算方法,得到:
[[a, a], [a]], [[b, b]]

这个结果貌似是不对的

上边的例子也过不了这个check:

CHECK_EQ(dataInput.getNumSequences(), shapeInput.getNumSequences());

因为:
dataInput.getNumSequences() == 2
shapeInput.getNumSequences() == 3

是不是应该改成:

CHECK_EQ(dataInput.getBatchSize(), shapeInput.getNumSequences());

Metadata

Metadata

Assignees

Labels

No labels
No labels

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions