Akasan
/
TorchUtils
mirror of https://github.com/Akasan/TorchUtils.git


  
1

	
2

	
3

	
4

	
5

	
6

	
7

	
8

	
9

	
10

	
            from torch import Tensor
from torch import nn
from typing import List, Any


def calculate_listed_layer(layers: List[Any], x: Tensor) -> Tensor:
    for layer in layers:
        x = layer(x)

    return x