publicUnclaimed/

gsnet

Clone

Design files

Lei KeUpdate README.mdb150d1354

This file is pinned to the selected release. Download the original to open it in your design tool.

Source

import torch
from torch import nn


class Scale(nn.Module):
    def __init__(self, init_value=1.0):
        super(Scale, self).__init__()
        self.scale = nn.Parameter(torch.FloatTensor([init_value]))

    def forward(self, input):
        return input * self.scale