@@ -24,7 +24,7 @@ def tile(input: Tensor, kernel: Tuple[int, int]) -> Tuple[Tensor, int, int]:
24
24
assert height % kh == 0
25
25
assert width % kw == 0
26
26
# TODO: Implement for Task 4.3.
27
- raise NotImplementedError (' Need to implement for Task 4.3' )
27
+ raise NotImplementedError (" Need to implement for Task 4.3" )
28
28
29
29
30
30
def avgpool2d (input : Tensor , kernel : Tuple [int , int ]) -> Tensor :
@@ -40,7 +40,7 @@ def avgpool2d(input: Tensor, kernel: Tuple[int, int]) -> Tensor:
40
40
"""
41
41
batch , channel , height , width = input .shape
42
42
# TODO: Implement for Task 4.3.
43
- raise NotImplementedError (' Need to implement for Task 4.3' )
43
+ raise NotImplementedError (" Need to implement for Task 4.3" )
44
44
45
45
46
46
max_reduce = FastOps .reduce (operators .max , - 1e9 )
@@ -68,13 +68,13 @@ class Max(Function):
68
68
def forward (ctx : Context , input : Tensor , dim : Tensor ) -> Tensor :
69
69
"Forward of max should be max reduction"
70
70
# TODO: Implement for Task 4.4.
71
- raise NotImplementedError (' Need to implement for Task 4.4' )
71
+ raise NotImplementedError (" Need to implement for Task 4.4" )
72
72
73
73
@staticmethod
74
74
def backward (ctx : Context , grad_output : Tensor ) -> Tuple [Tensor , float ]:
75
75
"Backward of max should be argmax (see above)"
76
76
# TODO: Implement for Task 4.4.
77
- raise NotImplementedError (' Need to implement for Task 4.4' )
77
+ raise NotImplementedError (" Need to implement for Task 4.4" )
78
78
79
79
80
80
def max (input : Tensor , dim : int ) -> Tensor :
@@ -97,7 +97,7 @@ def softmax(input: Tensor, dim: int) -> Tensor:
97
97
softmax tensor
98
98
"""
99
99
# TODO: Implement for Task 4.4.
100
- raise NotImplementedError (' Need to implement for Task 4.4' )
100
+ raise NotImplementedError (" Need to implement for Task 4.4" )
101
101
102
102
103
103
def logsoftmax (input : Tensor , dim : int ) -> Tensor :
@@ -116,7 +116,7 @@ def logsoftmax(input: Tensor, dim: int) -> Tensor:
116
116
log of softmax tensor
117
117
"""
118
118
# TODO: Implement for Task 4.4.
119
- raise NotImplementedError (' Need to implement for Task 4.4' )
119
+ raise NotImplementedError (" Need to implement for Task 4.4" )
120
120
121
121
122
122
def maxpool2d (input : Tensor , kernel : Tuple [int , int ]) -> Tensor :
@@ -132,7 +132,7 @@ def maxpool2d(input: Tensor, kernel: Tuple[int, int]) -> Tensor:
132
132
"""
133
133
batch , channel , height , width = input .shape
134
134
# TODO: Implement for Task 4.4.
135
- raise NotImplementedError (' Need to implement for Task 4.4' )
135
+ raise NotImplementedError (" Need to implement for Task 4.4" )
136
136
137
137
138
138
def dropout (input : Tensor , rate : float , ignore : bool = False ) -> Tensor :
@@ -148,4 +148,4 @@ def dropout(input: Tensor, rate: float, ignore: bool = False) -> Tensor:
148
148
tensor with random positions dropped out
149
149
"""
150
150
# TODO: Implement for Task 4.4.
151
- raise NotImplementedError (' Need to implement for Task 4.4' )
151
+ raise NotImplementedError (" Need to implement for Task 4.4" )
0 commit comments