Skip to content

GitLab

Verified Commit df54b73f authored by Max Ehrlich's avatar Max Ehrlich
Browse files

Import missing tensor definitions

parent 9bd45732
Hide whitespace changes
Inline Side-by-side
jpeg_layers/relu.py
View file @ df54b73f
import torch
import opt_einsum as oe
from jpeg_codec.tensors import D_n, D, Z, S_i, S
class JpegRelu(torch.nn.modules.Module):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment