Practical Pipelining for pyTorch
Project description
Torch Tensor Types
This package is a Quality of Life improvement when prototyping and processing Tensor objects from the pyTorch library. The TensorType class is a Pipeline for preprocessing tensors automatically, and include multiple utility methods. You can add TensorTypes together to have a longer preprocessing pipeline.
For example, this code
fake_image = model(torch.unsqueeze(real_image, 0).cuda()).cpu().detach().numpy()[0]
can be replaced by
fake_image = SingleDisplayableImage<<model(ModelInputFormat<<real_image)
Project details
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
Close
Hashes for torch_tensor_type-0.0.2-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 95cd38f425ca735c7932920337de245576f1c2694a5568aff9a08b4a5d76f351 |
|
MD5 | c33427f5a2c061968297f3c40210a33e |
|
BLAKE2b-256 | a3cf2e8af1d3856d6efa99d1b0ffd4b2616f36a6e9c2734f1adc9d2ae120ff81 |