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

Update masking.py

main

BBing 2 years ago

parent

5c4ab55e86

commit

eb394976a2

1 changed files with 0 additions and 1 deletions

Unified View

							
								+ 0
								
								- 1
							
utils/masking.pyView File
								
																@@ -2,7 +2,6 @@ import mindspore
															
																import mindspore.ops.operations as P
																
																import mindspore.ops.operations as P
															
																import mindspore.numpy as mnp
																
																import mindspore.numpy as mnp
															
																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]