from torch import nn from cape import CAPE1d pos_emb = CAPE1d(d_model=512) ... new_axis=100) positions_delta[0, :] *= torch.arange(1, ...
確定! 回上一頁