hc/JSCC
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update model.py

Browse Source
This commit is contained in:
ZHENG Chunhang 2024-01-16 23:02:10 +08:00 committed by GitHub
parent cb6e52e183
commit ab409da89e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
model.py
View File

@ -95,7 +95,7 @@ class _Encoder(nn.Module):
k = torch.prod(torch.tensor(z_hat.size())) k = torch.prod(torch.tensor(z_hat.size()))
else: else:
raise Exception('Unknown size of input') raise Exception('Unknown size of input')
k = torch.tensor(k) # k = torch.tensor(k)
z_temp = z_hat.reshape(batch_size, 1, 1, -1) z_temp = z_hat.reshape(batch_size, 1, 1, -1)
z_trans = z_hat.reshape(batch_size, 1, -1, 1) z_trans = z_hat.reshape(batch_size, 1, -1, 1)
tensor = torch.sqrt(P * k) * z_hat / torch.sqrt((z_temp @ z_trans)) tensor = torch.sqrt(P * k) * z_hat / torch.sqrt((z_temp @ z_trans))