Skip to content

Commit

Permalink
minor formatting fixes to README
Browse files Browse the repository at this point in the history
  • Loading branch information
egodfrey1618 committed Feb 10, 2023
1 parent 9be5903 commit 31531a2
Showing 1 changed file with 4 additions and 19 deletions.
23 changes: 4 additions & 19 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -706,19 +706,13 @@ def repeat(a: TT["i"], d: TT[1]) -> TT["d", "i"]:
raise NotImplementedError

test_repeat = make_test("repeat", repeat, repeat_spec, constraint=constraint_set)


# ## Puzzle 21 - bucketize
#
# Compute [bucketize](https://pytorch.org/docs/stable/generated/torch.bucketize.html)
```



![svg](Tensor%20Puzzlers_files/Tensor%20Puzzlers_68_0.svg)


## Puzzle 21 - bucketize

Compute [bucketize](https://pytorch.org/docs/stable/generated/torch.bucketize.html)

```python
def bucketize_spec(v, boundaries, out):
Expand All @@ -741,20 +735,11 @@ def bucketize(v: TT["i"], boundaries: TT["j"]) -> TT["i"]:

test_bucketize = make_test("bucketize", bucketize, bucketize_spec,
constraint=constraint_set)


#
# # Speed Run Mode!
#
# What is the smallest you can make each of these?
```



![svg](Tensor%20Puzzlers_files/Tensor%20Puzzlers_69_0.svg)



## Speed Run Mode!
What is the smallest you can make each of these?

```python
import inspect
Expand Down

0 comments on commit 31531a2

Please sign in to comment.