Source code for sinabs.layers.quantize
import torch.nn as nn
from sinabs.activation import Quantize
[docs]class QuantizeLayer(nn.Module):
"""
Layer that quantizes the input, i.e. returns floor(input).
:param quantize: If False, this layer will do nothing.
"""
def __init__(self, quantize=True):
super().__init__()
self.quantize = quantize