MyConcatenate Class Reference

Inheritance diagram for MyConcatenate:

Collaboration diagram for MyConcatenate:

Public Member Functions
	call (self, inputs)
	compute_mask (self, inputs, mask=None)
	get_config (self)

Detailed Description

Concatenate the 3D input tensors and their 2D masks along the axis=1 dimension.

Definition at line 15 of file model.py.

Member Function Documentation

call()

call	(		self,
			inputs )

Expect the input to be list of 3D tensors

Definition at line 18 of file model.py.

    def call(self, inputs):
        '''
        Expect the input to be list of 3D tensors
        '''
        return keras.ops.concatenate(inputs, axis=1)
 

compute_mask()

compute_mask	(		self,
			inputs,
			mask = None )

Expect the mask to be list of 2D mask tensors

Definition at line 24 of file model.py.

    def compute_mask(self, inputs, mask=None):
        '''
        Expect the mask to be list of 2D mask tensors
        '''
        if mask is None:
            return None
        return keras.ops.concatenate(mask, axis=1)
 

get_config()

get_config

(

self

)

Keras needs this

Definition at line 32 of file model.py.

    def get_config(self):
        '''
        Keras needs this
        '''
        return super().get_config()
 
 

---

The documentation for this class was generated from the following file:

• analysis/scripts/tflat/model.py