This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Pretty print tables summarizing properties of tensor arrays in numpy, pytorch, jax, etc. | |
| Now on pip! `pip install arrgh` https://github.com/nmwsharp/arrgh |
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| class DepthToSpace(nn.Module): | |
| def __init__(self, block_size): | |
| super(DepthToSpace, self).__init__() | |
| self.block_size = block_size | |
| self.block_size_sq = block_size*block_size | |
| def forward(self, input): | |
| output = input.permute(0, 2, 3, 1) | |
| (batch_size, d_height, d_width, d_depth) = output.size() | |
| s_depth = int(d_depth / self.block_size_sq) |
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| #include <iostream> | |
| #include <chrono> | |
| #include <ctime> | |
| #include <cmath> | |
| class Timer | |
| { | |
| public: | |
| void start() | |
| { |