bugfix: div-->dim (#18135)

This commit is contained in:
orgoro 2022-07-19 11:24:56 +03:00 committed by GitHub
parent e630dad555
commit 798384467b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -2181,7 +2181,7 @@ class PerceiverBasicDecoder(PerceiverAbstractDecoder):
if self.concat_preprocessed_input:
if inputs_without_pos is None:
raise ValueError("Value is required for inputs_without_pos if concat_preprocessed_input is True")
pos_emb = torch.cat([inputs_without_pos, pos_emb], div=-1)
pos_emb = torch.cat([inputs_without_pos, pos_emb], dim=-1)
return pos_emb