limingjuan
/
Prediction-MindSpore
Not watched
1
0
Fork 0
Code
Releases 0 Wiki evaluate Activity
Issues 0 Pull Requests 0 Datasets Model Cloudbrain HPC
Browse Source

Update masking.py

main
BBing 2 years ago
parent
bf81afc17d
commit
65f0551db0
1 changed files with 1 additions and 1 deletions
Split View
Diff Options
Show Stats Download Patch File Download Diff File
  1. +1
    -1
      utils/masking.py

+ 1
- 1
utils/masking.py View File

@@ -1,7 +1,7 @@
import mindspore
import mindspore.ops.operations as P
import mindspore.numpy as mnp
1
class TriangularCausalMask():
def __init__(self, B, L, device="cpu"):
mask_shape = [B, 1, L, L]


Write Preview
Loading…
Cancel
Save