Update masking.py · bf81afc17d - Prediction-MindSpore - 开源协同云脑生态支撑系统

Update masking.py

main

BBing 2 years ago

parent

f7a54667eb

commit

bf81afc17d

1 changed files with 1 additions and 1 deletions

Unified View

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