What's the second argument of puzzle 8 for?

srush / Tensor-Puzzles

Solve puzzles. Improve your pytorch.

MIT License

3.19k stars 269 forks source link

What's the second argument of puzzle 8 for? #13

Closed austinvhuang closed 2 years ago

austinvhuang commented 2 years ago

Puzzle 8 has two arguments:

def diff(a: TT["i"], i: int) -> TT["i"]:

What's the second int argument for? It seems like the operation would be fully specified for the first argument alone given shape is an allowed operation.

srush commented 2 years ago

It's not necessary. Just included so you can save characters.